Update multipurpose_chatbot/engines/transformers_engine.py

multipurpose_chatbot/engines/transformers_engine.py

@@ -546,7 +546,7 @@ class TransformersEngine(BaseEngine):
                     if message_safety is not None:
                         raise gr.Error(message_safety)
 
-    @maybe_spaces_gpu
+    # @maybe_spaces_gpu
     def generate_yield_string(self, prompt, temperature, max_tokens, stop_strings: Optional[Tuple[str]] = None, **kwargs):
         
         # ! MUST PUT INSIDE torch.no_grad() otherwise it will overflow OOM