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:
|