Spaces:
Runtime error
Runtime error
update that import spaces at top
Browse files- model_worker.py +1 -3
model_worker.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
"""
|
| 2 |
A model worker executes the model.
|
| 3 |
"""
|
|
|
|
| 4 |
import argparse
|
| 5 |
import asyncio
|
| 6 |
import json
|
|
@@ -29,9 +30,6 @@ from transformers import TextIteratorStreamer
|
|
| 29 |
from threading import Thread
|
| 30 |
from icecream import ic
|
| 31 |
|
| 32 |
-
import spaces # for use zero of huggingface
|
| 33 |
-
|
| 34 |
-
|
| 35 |
GB = 1 << 30
|
| 36 |
|
| 37 |
worker_id = str(uuid.uuid4())[:6]
|
|
|
|
| 1 |
"""
|
| 2 |
A model worker executes the model.
|
| 3 |
"""
|
| 4 |
+
import spaces # for use zero of huggingface
|
| 5 |
import argparse
|
| 6 |
import asyncio
|
| 7 |
import json
|
|
|
|
| 30 |
from threading import Thread
|
| 31 |
from icecream import ic
|
| 32 |
|
|
|
|
|
|
|
|
|
|
| 33 |
GB = 1 << 30
|
| 34 |
|
| 35 |
worker_id = str(uuid.uuid4())[:6]
|