pmolchanov commited on
Commit
897faea
·
verified ·
1 Parent(s): deed901

Update app_chat.py

Browse files
Files changed (1) hide show
  1. 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)