Update model.py
Browse files
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/
|
24 |
model = AutoModelForCausalLM.from_pretrained(
|
25 |
-
"MikeMpapa/lmd-4bars-2048-
|
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
|