Spaces:
Running
on
Zero
Running
on
Zero
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]
|