Archisman Karmakar
commited on
Commit
·
4b7388b
1
Parent(s):
2d6564f
Update model3.py
Browse files
transformation_and_Normalization/hmv_cfg_base_stage3/model3.py
CHANGED
@@ -42,7 +42,7 @@ model_class = globals()[model_info["model_class"]]
|
|
42 |
def load_model():
|
43 |
tokenizer = tokenizer_class.from_pretrained(hf_location)
|
44 |
print("Loading model 3")
|
45 |
-
model = model_class.from_pretrained(hf_location,
|
46 |
# device_map=torch.device(
|
47 |
# "cuda" if torch.cuda.is_available() else "cpu")
|
48 |
)
|
|
|
42 |
def load_model():
|
43 |
tokenizer = tokenizer_class.from_pretrained(hf_location)
|
44 |
print("Loading model 3")
|
45 |
+
model = model_class.from_pretrained(hf_location, use_safetensors=True,
|
46 |
# device_map=torch.device(
|
47 |
# "cuda" if torch.cuda.is_available() else "cpu")
|
48 |
)
|