exact-railcar commited on
Commit
20d2f51
·
verified ·
1 Parent(s): 1716635

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -35,6 +35,7 @@ unet = animatediff.models.unet.UNet2DConditionModel.from_pretrained_2d(
35
  subfolder='unet',
36
  unet_additional_kwargs=infer_config.unet_additional_kwargs,
37
  feature_extractor = transformers.CLIPImageProcessor.from_pretrained(base_model, subfolder='feature_extractor')
 
38
 
39
  pipeline = diffusers.StableDiffusionPipeline.from_single_file('https://huggingface.co/chaowenguoback/15/blob/main/chilloutMix-Ni.safetensors', config='stable-diffusion-v1-5/stable-diffusion-v1-5', safety_checker=None, use_safetensors=True)
40
  unet.load_state_dict(pipeline.unet.state_dict(), strict=False)
 
35
  subfolder='unet',
36
  unet_additional_kwargs=infer_config.unet_additional_kwargs,
37
  feature_extractor = transformers.CLIPImageProcessor.from_pretrained(base_model, subfolder='feature_extractor')
38
+ )
39
 
40
  pipeline = diffusers.StableDiffusionPipeline.from_single_file('https://huggingface.co/chaowenguoback/15/blob/main/chilloutMix-Ni.safetensors', config='stable-diffusion-v1-5/stable-diffusion-v1-5', safety_checker=None, use_safetensors=True)
41
  unet.load_state_dict(pipeline.unet.state_dict(), strict=False)