Update multipurpose_chatbot/engines/sealmmm_engine.py

multipurpose_chatbot/engines/sealmmm_engine.py

@@ -4,6 +4,7 @@
 import os
 import numpy as np
 import argparse
+import spaces
 import torch
 import gradio as gr
 from typing import Any, Iterator
@@ -201,7 +202,8 @@ class SeaLMMMv0Engine(TransformersEngine):
                     message_safety = safety_check_conversation_string(message)
                     if message_safety is not None:
                         raise gr.Error(message_safety)
-    
+
+    @spaces.GPU
     def generate_yield_string(self, prompt, temperature, max_tokens, stop_strings: Optional[Tuple[str]] = None, **kwargs):
         from transformers.generation.utils import GenerationConfig
         from PIL import Image