Hugging Face's logo

Spaces:
HemaAM
/
GPT_train_on_LLaMa
Running

App Files Community
HemaAM commited on
Commit
b32c86f
·
1 Parent(s): 185e54c

Changed the trained model name

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -33,7 +33,7 @@ examples = [
33
  model_name = "pythia-160m"
34
  name = "redpajama"
35
 
36
- checkpoint_dir = Path("iter-010915-ckpt.pth")
37
  quantize = None
38
  strategy = "auto"
39
  devices = 1
 
33
  model_name = "pythia-160m"
34
  name = "redpajama"
35
 
36
+ checkpoint_dir = Path("iter-014870-ckpt.pth")
37
  quantize = None
38
  strategy = "auto"
39
  devices = 1