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-
|
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")
|
24 |
model = AutoModelForCausalLM.from_pretrained(
|
25 |
+
"MikeMpapa/lmd-8bars-2048-epochs10-testrun"
|
26 |
)
|
27 |
|
28 |
# Move model to device
|