new

pipeline/kiss3d_wrapper.py

@@ -257,7 +257,7 @@ class kiss3d_wrapper(object):
         caption_text = self.get_detailed_prompt(caption_text)
 
         return caption_text
-
+    @spaces.GPU
     def get_detailed_prompt(self, prompt, seed=None):
         if self.llm_model is not None:
             detailed_prompt = get_llm_response(self.llm_model, self.llm_tokenizer, prompt, seed=seed)