frozenleaves/trl
1
0
Fork 0
mirror of https://github.com/huggingface/trl.git synced 2025-10-20 18:43:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Comment Dbrx

Browse Source
This commit is contained in:
Quentin Gallouédec
2025-10-06 00:29:34 +00:00
parent 1eff7da9e0
commit ee8e6a7ba3
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_data_utils.py
View File

@ -284,7 +284,7 @@ class IsConversationalFromValueTester(TrlTestCase):
class ApplyChatTemplateTester(TrlTestCase): class ApplyChatTemplateTester(TrlTestCase):
tokenizers = [ tokenizers = [
"trl-internal-testing/tiny-CohereForCausalLM", "trl-internal-testing/tiny-CohereForCausalLM",
"trl-internal-testing/tiny-DbrxForCausalLM", # "trl-internal-testing/tiny-DbrxForCausalLM",
"trl-internal-testing/tiny-DeepseekV3ForCausalLM", "trl-internal-testing/tiny-DeepseekV3ForCausalLM",
"trl-internal-testing/tiny-DeepseekV3ForCausalLM-0528", "trl-internal-testing/tiny-DeepseekV3ForCausalLM-0528",
"trl-internal-testing/tiny-FalconMambaForCausalLM", "trl-internal-testing/tiny-FalconMambaForCausalLM",

2
tests/test_modeling_value_head.py
View File

@ -27,7 +27,7 @@ from .testing_utils import TrlTestCase
ALL_CAUSAL_LM_MODELS = [ ALL_CAUSAL_LM_MODELS = [
"trl-internal-testing/tiny-BloomForCausalLM", "trl-internal-testing/tiny-BloomForCausalLM",
"trl-internal-testing/tiny-CohereForCausalLM", "trl-internal-testing/tiny-CohereForCausalLM",
"trl-internal-testing/tiny-DbrxForCausalLM", # "trl-internal-testing/tiny-DbrxForCausalLM",
# "trl-internal-testing/tiny-FalconMambaForCausalLM", # FalconMambaForCausalLM modeling seems to be broken for now # "trl-internal-testing/tiny-FalconMambaForCausalLM", # FalconMambaForCausalLM modeling seems to be broken for now
"trl-internal-testing/tiny-Gemma2ForCausalLM", "trl-internal-testing/tiny-Gemma2ForCausalLM",
"trl-internal-testing/tiny-GemmaForCausalLM", "trl-internal-testing/tiny-GemmaForCausalLM",