horseee / DeepCache

[CVPR 2024] DeepCache: Accelerating Diffusion Models for Free
https://horseee.github.io/Diffusion_DeepCache/
Apache License 2.0
767 stars 36 forks source link

It is not work when using the base model in safetensors format. #17

Open l-dawei opened 9 months ago

l-dawei commented 9 months ago

When I use the basic model in safetensors format, the error is as follows:

T60E5dvnjQ

Who can help me?

horseee commented 9 months ago

Hi @l-dawei,

Could you check which UNet2DConditionModel the code would call? The one in diffusers or the one in DeepCache/sdxl/unet_2d_condition.py

l-dawei commented 9 months ago

Hi @l-dawei,

Could you check which UNet2DConditionModel the code would call? The one in diffusers or the one in DeepCache/sdxl/unet_2d_condition.py

I checked and found that if I use the from_single_file function to load, the UNet2DConditionModel of diffusers is still called.

l-dawei commented 9 months ago

Hi @l-dawei,

Could you check which UNet2DConditionModel the code would call? The one in diffusers or the one in DeepCache/sdxl/unet_2d_condition.py

When I use from_pretrained and from_single_file the different unet_2d_condition is called.

horseee commented 9 months ago

Hi @l-dawei,

Could you check if the new implementation works?

l-dawei commented 9 months ago

Hi @l-dawei,

Could you check if the new implementation works?

It's so powerful.