fix

gagan3012 · Apr 18, 2022 · 6f807b9 · 6f807b9
1 parent 73df143
commit 6f807b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trainer/main.py b/trainer/main.py
@@ -15,4 +15,4 @@
 model.from_pretrained(model_name="t5-base")
 model.train(train_df=train_df, test_df=eval_df, batch_size=4, max_epochs=10, use_gpu=False,tpu_cores=8)
 print(model.evaluate(test_df=test_df))
-model.upload(hf_username="gagan3012", model_name="k2tt5", token="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")
+model.upload(hf_username="gagan3012", model_name="k2tt5")