Spaces:
Runtime error
Runtime error
Plat
commited on
Commit
•
6277c6e
1
Parent(s):
eabde3c
feat: torch compile
Browse files
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|>"
|