작심삼일

[Diffusers 에러] OSError: Error no file named pytorch_model.bin, tf_model.h5, model.ckpt.index or flax_model.msgpack found in directory 본문

정리/잡 에러들

[Diffusers 에러] OSError: Error no file named pytorch_model.bin, tf_model.h5, model.ckpt.index or flax_model.msgpack found in directory

yun_s 2023. 6. 23. 21:27
728x90
반응형

에러

pipe = StableDiffusionControlNetPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16)

위처럼 Diffusers에서 StableDiffusion을 사용할 때 아래와 같은 에러를 만날 수 있다.

 

OSError: Error no file named pytorch_model.bin, tf_model.h5, model.ckpt.index or flax_model.msgpack found in directory /root/.cache/huggingface/hub/models--runwayml--stable-diffusion-v1-5/snapshots/aa9ba505e1973ae5cd05f5aedd345178f52f8e6a/text_encoder.

해결 방법

Transformers 버전을 4.26.0으로 다시 깔아준다.

pip install transformers==4.26.0
728x90
반응형
Comments