Spaces:
Runtime error
Runtime error
Update gradio_app.py
Browse files- gradio_app.py +4 -1
gradio_app.py
CHANGED
@@ -159,9 +159,12 @@ def get_image(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123, i
|
|
159 |
start = time.time()
|
160 |
gpu_id=0
|
161 |
if steps > 60:
|
162 |
-
steps = 60
|
|
|
|
|
163 |
model = model_list[gpu_id]
|
164 |
model = model.cuda()
|
|
|
165 |
batch_size=1
|
166 |
channels = model.model.diffusion_model.out_channels
|
167 |
frames = model.temporal_length
|
|
|
159 |
start = time.time()
|
160 |
gpu_id=0
|
161 |
if steps > 60:
|
162 |
+
steps = 60
|
163 |
+
|
164 |
+
global model_list
|
165 |
model = model_list[gpu_id]
|
166 |
model = model.cuda()
|
167 |
+
|
168 |
batch_size=1
|
169 |
channels = model.model.diffusion_model.out_channels
|
170 |
frames = model.temporal_length
|