Spaces:

gustavoaq
/

llama_7b_ft_2

Paused

gustavoaq commited on Apr 9, 2023

Commit

ae1f2b4

1 Parent(s): 25f93cb

Update finetune.py

Files changed (1) hide show

finetune.py CHANGED Viewed

@@ -155,7 +155,7 @@ trainer = transformers.Trainer(
         ddp_find_unused_parameters=False if ddp else None,
         push_to_hub_token="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",
         push_to_hub=True,
-        push_to_hub_model_id="gustavoaq/llama_ft"
     ),
     data_collator=transformers.DataCollatorForLanguageModeling(tokenizer, mlm=False),
 )

         ddp_find_unused_parameters=False if ddp else None,
         push_to_hub_token="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",
         push_to_hub=True,
+        push_to_hub_model_id="llama_ft"
     ),
     data_collator=transformers.DataCollatorForLanguageModeling(tokenizer, mlm=False),
 )