Back to search
Top Contributors
17
commits
Stars
2
Forks
0
Watchers
2
Open Issues
0
Overall repository health assessment
No package.json found
This might not be a Node.js project
17
commits
chore: Update states_generator.py to enable CUDA support for torch.autocast
7b78c38View on GitHubchore: Update fill_mask_api.py to use 'cuda' if device starts with 'cuda' for torch.autocast
751e97eView on GitHubchore: Update CUDA support and model file path in test_sentence_embedding_cross_encoder.py
717f099View on GitHubchore: Update test_sentence_embedding_cross_encoder.py with CUDA support and model file path
dc1e1a9View on GitHubchore: Update test_states_generator.py and test_mlm.py, and add rwkv6_op.cpp for CUDA support
10abee6View on GitHubchore: Refactor model_encoder_run.py to return logits in forward method
ad11172View on GitHubchore: Update test_states_generator.py and test_sentence_embedding.py, and refactor utilities.py and model_encoder_run.py
0af5efdView on GitHubchore: Add StatesGenerator class for generating states and text output
b5eac72View on GitHubchore: Refactor model_encoder_run.py to return logits in forward method
e0b6eaaView on GitHubchore: Refactor model_encoder_run.py and add test_sentence_embedding.py
b5bb09cView on GitHub