add space

tinychart/model/language_model/llava_phi.py

@@ -29,6 +29,7 @@ from transformers.modeling_outputs import CausalLMOutputWithPast
 
 from tinychart.model.llava_arch import LlavaMetaModel, LlavaMetaForCausalLM
 from tinychart.model.model_factory import *
+import spaces
 
 class TinyChartPhiConfig(PhiConfig):
     model_type = "tiny_chart_phi"
@@ -103,6 +104,7 @@ class TinyChartPhiForCausalLM(PhiForCausalLM, LlavaMetaForCausalLM):
         )
 
     @torch.no_grad()
+    @spaces.GPU
     def generate(
         self,
         inputs: Optional[torch.Tensor] = None,