abot-world-interactive

Paused

multimodalart HF Staff commited on Jul 12

Commit

9563308

verified ·

1 Parent(s): ab92f92

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -337,7 +337,7 @@ def create_gpu_rollout_loop(command_queue, seed_path, prompt_text, seed):
     Only picklable primitives + the multiprocessing `command_queue` cross the
     fork boundary. Live controls (held key set) and stop arrive via that queue.
     """
-    @spaces.GPU(duration=GPU_DURATION)
     def gpu_rollout():
         device = torch.device("cuda")
         prompt = (prompt_text or DEFAULT_PROMPT).strip() or DEFAULT_PROMPT

     Only picklable primitives + the multiprocessing `command_queue` cross the
     fork boundary. Live controls (held key set) and stop arrive via that queue.
     """
+    @spaces.GPU(duration=GPU_DURATION, size="xlarge")
     def gpu_rollout():
         device = torch.device("cuda")
         prompt = (prompt_text or DEFAULT_PROMPT).strip() or DEFAULT_PROMPT