xzl12306 commited on
Commit
6e58b78
·
verified ·
1 Parent(s): b733be1

update that import spaces at top

Browse files
Files changed (1) hide show
  1. 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]