Spaces:
Runtime error
Runtime error
Update inference.py
Browse files- inference.py +1 -0
inference.py
CHANGED
@@ -21,6 +21,7 @@ from threading import Thread
|
|
21 |
from typing import List
|
22 |
|
23 |
import torch
|
|
|
24 |
import transformers
|
25 |
from transformers import (
|
26 |
AutoModelForCausalLM,
|
|
|
21 |
from typing import List
|
22 |
|
23 |
import torch
|
24 |
+
torch.jit.script = lambda f: f
|
25 |
import transformers
|
26 |
from transformers import (
|
27 |
AutoModelForCausalLM,
|