amazonaws-la commited on
Commit
8b748cc
·
verified ·
1 Parent(s): 6064cea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -61,7 +61,6 @@ def generate(
61
 
62
  if not use_vae:
63
  pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
64
- sampler_class = sampler
65
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
66
 
67
  if use_vae:
 
61
 
62
  if not use_vae:
63
  pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
 
64
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
65
 
66
  if use_vae: