Shuang59 commited on
Commit
38930b8
β€’
1 Parent(s): 3f1a935

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -256,7 +256,7 @@ def compose_clevr_objects(prompt):
256
  )
257
 
258
  def sample(coordinates):
259
- samples = diffusion.p_sample_loop(
260
  model_fn,
261
  (len(coordinates), 3, options["image_size"], options["image_size"]),
262
  device=device,
 
256
  )
257
 
258
  def sample(coordinates):
259
+ samples = clevr_diffusion.p_sample_loop(
260
  model_fn,
261
  (len(coordinates), 3, options["image_size"], options["image_size"]),
262
  device=device,