MikeMpapa commited on
Commit
35d3880
·
verified ·
1 Parent(s): 2a0ab09

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -3
model.py CHANGED
@@ -20,10 +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_artist", use_auth_token=True)
24
  model = AutoModelForCausalLM.from_pretrained(
25
- "MikeMpapa/lmd-4bars-2048-epochs7",
26
- use_auth_token=True
27
  )
28
 
29
  # 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