Yaron Koresh commited on
Commit
5c32136
·
verified ·
1 Parent(s): 0adc98a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -110,13 +110,13 @@ function custom(){
110
 
111
  image_pipe = StableDiffusionPipeline.from_pretrained(base, torch_dtype=dtype, safety_checker=None).to(device)
112
  pipe = AnimateDiffPipeline.from_pretrained(base, torch_dtype=dtype, motion_adapter=adapter).to(device)
113
- pipe.scheduler = DDIMScheduler(
 
 
114
  clip_sample=False,
115
- beta_start=0.00085,
116
- beta_end=0.012,
117
  beta_schedule="linear",
118
- timestep_spacing="trailing",
119
- steps_offset=1
120
  )
121
  pipe.enable_vae_slicing()
122
  pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter_sd15.bin")
 
110
 
111
  image_pipe = StableDiffusionPipeline.from_pretrained(base, torch_dtype=dtype, safety_checker=None).to(device)
112
  pipe = AnimateDiffPipeline.from_pretrained(base, torch_dtype=dtype, motion_adapter=adapter).to(device)
113
+ pipe.scheduler = DDIMScheduler.from_pretrained(
114
+ base,
115
+ subfolder="scheduler",
116
  clip_sample=False,
117
+ timestep_spacing="linspace",
 
118
  beta_schedule="linear",
119
+ steps_offset=1,
 
120
  )
121
  pipe.enable_vae_slicing()
122
  pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter_sd15.bin")