search

AiuniAI / Unique3D

[NeurIPS 2024] Unique3D: High-Quality and Efficient 3D Mesh Generation from a Single Image
https://wukailu.github.io/Unique3D/
MIT License
3.1k stars 246 forks source link

StableDiffusionControlNetPipeline loading stuck #104

Open SensenGao opened 2 months ago

SensenGao commented 2 months ago

image

Has anyone encountered this issue where loading the StableDiffusionControlNetPipeline gets stuck? The first time, loading the StableDiffusionPipeline works fine, but the second time, when the StableDiffusionControlNetPipeline calls the function model_cls.from_pretrained(ckpt_or_pretrained, torch_dtype=torch_dtype, **kwargs), it gets stuck at this point and remains so for several hours without progressing.

josepmy commented 2 months ago

@SensenGao Did you check this? https://github.com/AiuniAI/Unique3D/issues/95

SensenGao commented 2 months ago

@SensenGao Did you check this? #95

I have already tried replacing it with "ZeroCool94/stable-diffusion-v1-5", but once it reaches pipe = model_cls.from_pretrained(ckpt_or_pretrained, torch_dtype=torch_dtype, **kwargs), i.e., StableDiffusionControlNetPipeline.from_pretrained, it gets stuck. There is no such issue when using StableDiffusionPipeline.

josepmy commented 2 months ago

@SensenGao try with base_model = "benjamin-paine/stable-diffusion-v1-5"

SensenGao commented 2 months ago

@SensenGao try with base_model = "benjamin-paine/stable-diffusion-v1-5"

Thank you for your reply. I will have a try.