Spaces:

chansung
/

vid2persona

Paused

chansung commited on Mar 11

Commit

8b86a0a

•

1 Parent(s): 3397c46

Update vid2persona/gen/local_openllm.py

Files changed (1) hide show

vid2persona/gen/local_openllm.py CHANGED Viewed

@@ -1,4 +1,4 @@
-import spaces
 import torch
 from threading import Thread
@@ -8,7 +8,7 @@ from transformers import TextIteratorStreamer
 model = None
 tokenizer = None
-@spaces.GPU
 def send_message(
     messages: list,
     model_id: str,

+# import spaces
 import torch
 from threading import Thread
 model = None
 tokenizer = None
+# @spaces.GPU
 def send_message(
     messages: list,
     model_id: str,