update(*): HF space support.

app.py

@@ -13,6 +13,7 @@ from einops import rearrange
 from datetime import datetime
 from typing import Optional, List, Dict
 from huggingface_hub import snapshot_download
+import spaces
 
 os.environ["GRADIO_TEMP_DIR"] = os.path.abspath(os.path.join(os.path.dirname(__file__), "gradio_cache"))
 
@@ -526,6 +527,7 @@ def validate_inputs(num_frames, num_cond_images, num_cond_sketches, text_prompt,
     
     return errors
 
+@spaces.GPU
 def tooncomposer_inference(num_frames, num_cond_images, num_cond_sketches, text_prompt, cfg_scale, sequence_cond_residual_scale, resolution, *args):
     # Validate inputs first
     validation_errors = validate_inputs(num_frames, num_cond_images, num_cond_sketches, text_prompt, *args)

pipeline/i2v_pipeline.py

@@ -160,6 +160,8 @@ class WanVideoPipeline(BasePipeline):
                 state_dict, config = state_dict
         config.update(config_dict or {})
         model = model_cls(**config)
+        if torch.cuda.is_available():
+            model = model.to("cuda")
         if "use_local_lora" in config_dict or "use_dera" in config_dict:
             strict = False
         missing_keys, unexpected_keys = model.load_state_dict(state_dict, strict=strict)