fede97 commited on
Commit
55e6f4a
1 Parent(s): aac6e8d

update checkpoints

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
11
 
12
  # training from scratch with latbert tokenizer
13
  CHECKPOINT_PATH= 'scratch_2-nodes_tokenizer_latbert-original_packing_fcocchi/'
14
- CHECKPOINT_PATH= 'itserr/latin_llm_alpha'
15
 
16
  print(f"Loading model from: {CHECKPOINT_PATH}")
17
  tokenizer = AutoTokenizer.from_pretrained(CHECKPOINT_PATH, token=os.environ['HF_TOKEN_READ'])
 
11
 
12
  # training from scratch with latbert tokenizer
13
  CHECKPOINT_PATH= 'scratch_2-nodes_tokenizer_latbert-original_packing_fcocchi/'
14
+ CHECKPOINT_PATH= 'itserr/scratch_2-nodes_tokenizer_latbert-original_packing_fcocchi'
15
 
16
  print(f"Loading model from: {CHECKPOINT_PATH}")
17
  tokenizer = AutoTokenizer.from_pretrained(CHECKPOINT_PATH, token=os.environ['HF_TOKEN_READ'])