Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|