Spaces:
Paused
Paused
Update app_chat.py
Browse files- app_chat.py +1 -1
app_chat.py
CHANGED
@@ -26,7 +26,7 @@ DESCRIPTION = """\
|
|
26 |
model_id = "nvidia/Hymba-1.5B-Instruct"
|
27 |
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="cuda", trust_remote_code=True)
|
28 |
model = model.cuda().to(torch.bfloat16)
|
29 |
-
|
30 |
#model.to('cuda')
|
31 |
#model.eval()
|
32 |
tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
|
|
|
26 |
model_id = "nvidia/Hymba-1.5B-Instruct"
|
27 |
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="cuda", trust_remote_code=True)
|
28 |
model = model.cuda().to(torch.bfloat16)
|
29 |
+
model.compile()
|
30 |
#model.to('cuda')
|
31 |
#model.eval()
|
32 |
tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
|