Spaces:
Runtime error
Runtime error
sarath-shekkizhar
commited on
Commit
•
4808ae9
1
Parent(s):
cc4fea0
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ if not torch.cuda.is_available():
|
|
30 |
if torch.cuda.is_available():
|
31 |
model_id = "tenyx/Llama3-TenyxChat-70B"
|
32 |
# model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.bfloat16)
|
33 |
-
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto",
|
34 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
35 |
tokenizer.use_default_system_prompt = False
|
36 |
|
|
|
30 |
if torch.cuda.is_available():
|
31 |
model_id = "tenyx/Llama3-TenyxChat-70B"
|
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 |
|