1inkusFace commited on
Commit
e6e4c71
·
verified ·
1 Parent(s): 98f4916

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
29
  predictor = None # Global predictor, BUT loaded inside a function
30
 
31
  def get_transformer_model_id(task_type: str) -> str:
32
- return "Skywork/SkyReels-V1-Hunyuan-I2V" if task_type == "i2v" else "Skywork/SkyReels-V1-Hunyuan-T2V"
33
 
34
  @spaces.GPU(duration=90)
35
  def init_predictor(task_type: str):
 
29
  predictor = None # Global predictor, BUT loaded inside a function
30
 
31
  def get_transformer_model_id(task_type: str) -> str:
32
+ return "Skywork/skyreels-v1-Hunyuan-i2v" if task_type == "i2v" else "Skywork/skyreels-v1-Hunyuan-t2v"
33
 
34
  @spaces.GPU(duration=90)
35
  def init_predictor(task_type: str):