MaxMilan1 commited on
Commit
82af72d
·
1 Parent(s): 449e4e2

@spaces.GPU?

Browse files
Files changed (1) hide show
  1. util/v3d.py +1 -1
util/v3d.py CHANGED
@@ -148,7 +148,7 @@ def change_model_params(model, min_cfg, max_cfg):
148
  model.sampler.guider.max_scale = max_cfg
149
  model.sampler.guider.min_scale = min_cfg
150
 
151
-
152
  def prep():
153
  model_config = "scripts/pub/configs/V3D_512.yaml"
154
  num_frames = OmegaConf.load(
 
148
  model.sampler.guider.max_scale = max_cfg
149
  model.sampler.guider.min_scale = min_cfg
150
 
151
+ @spaces.GPU
152
  def prep():
153
  model_config = "scripts/pub/configs/V3D_512.yaml"
154
  num_frames = OmegaConf.load(