MikeMpapa commited on
Commit
98a2140
·
verified ·
1 Parent(s): 3d27b95

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -20,9 +20,9 @@ def get_model_and_tokenizer() -> Tuple[AutoModelForCausalLM, AutoTokenizer]:
20
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
21
 
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
 
20
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
21
 
22
  # Load the tokenizer and the model
23
+ tokenizer = AutoTokenizer.from_pretrained("MikeMpapa/lmd_mmm_tokenizer_tutorial_artist")
24
  model = AutoModelForCausalLM.from_pretrained(
25
+ "MikeMpapa/lmd-4bars-2048-epochs7"
26
  )
27
 
28
  # Move model to device