juancopi81 commited on
Commit
4e53ed5
1 Parent(s): 735366b

Add get_model_tokenizer

Browse files
Files changed (1) hide show
  1. utils.py +1 -5
utils.py CHANGED
@@ -15,11 +15,7 @@ from model import get_model_and_tokenizer
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
 
17
  # Load the tokenizer and the model
18
- tokenizer = AutoTokenizer.from_pretrained("juancopi81/lmd_8bars_tokenizer")
19
- model = AutoModelForCausalLM.from_pretrained("juancopi81/lmd-8bars-2048-epochs20_v3")
20
-
21
- # Move model to device
22
- model = model.to(device)
23
 
24
 
25
  def create_seed_string(genre: str = "OTHER") -> str:
 
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
 
17
  # Load the tokenizer and the model
18
+ model, tokenizer = get_model_and_tokenizer()
 
 
 
 
19
 
20
 
21
  def create_seed_string(genre: str = "OTHER") -> str: