MikeMpapa commited on
Commit
3d27b95
·
verified ·
1 Parent(s): 7d40772

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -22,7 +22,7 @@ def get_model_and_tokenizer() -> Tuple[AutoModelForCausalLM, AutoTokenizer]:
22
  # Load the tokenizer and the model
23
  tokenizer = AutoTokenizer.from_pretrained("MikeMpapa/lmd_mmm_tokenizer_tutorial")
24
  model = AutoModelForCausalLM.from_pretrained(
25
- "MikeMpapa/lmd-8bars-2048-epochs10-testrun"
26
  )
27
 
28
  # Move model to device
 
22
  # Load the tokenizer and the model
23
  tokenizer = AutoTokenizer.from_pretrained("MikeMpapa/lmd_mmm_tokenizer_tutorial")
24
  model = AutoModelForCausalLM.from_pretrained(
25
+ "MikeMpapa/lmd-4bars-2048-epochs14"
26
  )
27
 
28
  # Move model to device