Plat commited on
Commit
6277c6e
1 Parent(s): eabde3c

feat: torch compile

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -27,6 +27,10 @@ model = AutoModelForCausalLM.from_pretrained(
27
  token=HF_READ_TOKEN,
28
  )
29
 
 
 
 
 
30
 
31
  BOS = "<|bos|>"
32
  EOS = "<|eos|>"
 
27
  token=HF_READ_TOKEN,
28
  )
29
 
30
+ try:
31
+ model = torch.compile(model)
32
+ except:
33
+ print("torch compile not supported")
34
 
35
  BOS = "<|bos|>"
36
  EOS = "<|eos|>"