trigger llama

huggingface · Feb 29, 2024 · 59ecf36 · 59ecf36
1 parent 317f044
commit 59ecf36
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/models/llama/test_llama.py b/tests/models/llama/test_llama.py
@@ -40,7 +40,8 @@
     output_dir="output",
     per_device_train_batch_size=1,
     gradient_accumulation_steps=2,
-    num_train_epochs=3,
+    # num_train_epochs=1,
+    max_steps=100,
     logging_strategy="steps",
     logging_steps=20,
     bf16=True,