Spaces:
Running
on
Zero
Running
on
Zero
Happzy-WHU
commited on
Commit
•
e862792
1
Parent(s):
f1964c8
v3 modified
Browse files
V3.py
CHANGED
|
@@ -8,6 +8,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 8 |
)
|
| 9 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 10 |
|
|
|
|
| 11 |
def api_call(messages):
|
| 12 |
text = tokenizer.apply_chat_template(
|
| 13 |
messages,
|
|
|
|
| 8 |
)
|
| 9 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 10 |
|
| 11 |
+
@spaces.GPU
|
| 12 |
def api_call(messages):
|
| 13 |
text = tokenizer.apply_chat_template(
|
| 14 |
messages,
|