Spaces:

HemaAM
/

GPT_train_on_LLaMa

Running

HemaAM commited on Nov 25, 2023

Commit

b32c86f

•

1 Parent(s): 185e54c

Changed the trained model name

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -33,7 +33,7 @@ examples = [
 model_name = "pythia-160m"
 name = "redpajama"
-checkpoint_dir = Path("iter-010915-ckpt.pth")
 quantize = None
 strategy = "auto"
 devices = 1

 model_name = "pythia-160m"
 name = "redpajama"
+checkpoint_dir = Path("iter-014870-ckpt.pth")
 quantize = None
 strategy = "auto"
 devices = 1