heheyas commited on
Commit
b9555e5
1 Parent(s): 582c5e4
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -147,6 +147,7 @@ def change_model_params(model, min_cfg, max_cfg):
147
  model.sampler.guider.min_scale = min_cfg
148
 
149
 
 
150
  def launch(device="cuda", port=4321, share=False):
151
  model_config = "scripts/pub/configs/V3D_512.yaml"
152
  num_frames = OmegaConf.load(
 
147
  model.sampler.guider.min_scale = min_cfg
148
 
149
 
150
+ @spaces.GPU
151
  def launch(device="cuda", port=4321, share=False):
152
  model_config = "scripts/pub/configs/V3D_512.yaml"
153
  num_frames = OmegaConf.load(