fbnnb commited on
Commit
02796fb
·
verified ·
1 Parent(s): b15fbc7

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. 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