Spaces:

shi-labs
/

CuMo-7b-zero

Running on Zero

jiachenl commited on May 10, 2024

Commit

360eb6f

verified ·

1 Parent(s): 5061a0e

Update cumo/serve/model_worker.py

Files changed (1) hide show

cumo/serve/model_worker.py CHANGED Viewed

@@ -14,7 +14,7 @@ import requests
 import torch
 import uvicorn
 from functools import partial
-import spaces
 from cumo.constants import WORKER_HEART_BEAT_INTERVAL
 from cumo.utils import (build_logger, server_error_msg,

 import torch
 import uvicorn
 from functools import partial
+#import spaces
 from cumo.constants import WORKER_HEART_BEAT_INTERVAL
 from cumo.utils import (build_logger, server_error_msg,