Spaces:
Sleeping
Sleeping
xavierbarbier
commited on
Commit
•
9707f42
1
Parent(s):
93c3e30
Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
37 |
|
38 |
model_name = "microsoft/Phi-3.5-mini-instruct"
|
39 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
40 |
-
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.
|
41 |
|
42 |
print("Start the model init process")
|
43 |
"""model = model = GPT4All(model_name, model_path, allow_download = False, device="cpu")
|
|
|
37 |
|
38 |
model_name = "microsoft/Phi-3.5-mini-instruct"
|
39 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
40 |
+
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
|
41 |
|
42 |
print("Start the model init process")
|
43 |
"""model = model = GPT4All(model_name, model_path, allow_download = False, device="cpu")
|