Update utils.py
Browse files
utils.py
CHANGED
|
@@ -80,6 +80,7 @@ tokenizer = AutoTokenizer.from_pretrained("dbmdz/bert-base-italian-cased")
|
|
| 80 |
model = AutoModelForCausalLM.from_pretrained("dbmdz/bert-base-italian-cased")
|
| 81 |
|
| 82 |
def load_tokenizer_and_model(base_model,load_8bit=False):
|
|
|
|
| 83 |
if torch.cuda.is_available():
|
| 84 |
device = "cuda"
|
| 85 |
else:
|
|
|
|
| 80 |
model = AutoModelForCausalLM.from_pretrained("dbmdz/bert-base-italian-cased")
|
| 81 |
|
| 82 |
def load_tokenizer_and_model(base_model,load_8bit=False):
|
| 83 |
+
base_model = "dbmdz/bert-base-italian-cased"
|
| 84 |
if torch.cuda.is_available():
|
| 85 |
device = "cuda"
|
| 86 |
else:
|