Spaces:
Runtime error
Runtime error
juancopi81
commited on
Commit
•
4e53ed5
1
Parent(s):
735366b
Add get_model_tokenizer
Browse files
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 =
|
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:
|