MaxMilan1
commited on
Commit
·
82af72d
1
Parent(s):
449e4e2
@spaces.GPU?
Browse files- 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(
|