Spaces:
Runtime error
Runtime error
Romain-Cosentino
commited on
Commit
•
e870d99
1
Parent(s):
70ef319
Update app.py
Browse files
app.py
CHANGED
@@ -29,8 +29,8 @@ if not torch.cuda.is_available():
|
|
29 |
|
30 |
if torch.cuda.is_available():
|
31 |
model_id = "tenyx/TenyxChat-8x7B-v1"
|
32 |
-
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.bfloat16)
|
33 |
-
|
34 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
35 |
tokenizer.use_default_system_prompt = False
|
36 |
|
|
|
29 |
|
30 |
if torch.cuda.is_available():
|
31 |
model_id = "tenyx/TenyxChat-8x7B-v1"
|
32 |
+
#model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.bfloat16)
|
33 |
+
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", load_in_4bit=True)
|
34 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
35 |
tokenizer.use_default_system_prompt = False
|
36 |
|