search

deep-floyd / IF

Other
7.69k stars 501 forks source link

demo code on huggingface is not runnable #78

Open jemmyshin opened 1 year ago

jemmyshin commented 1 year ago

I got error when running the sample code on hf: https://huggingface.co/docs/diffusers/api/pipelines/if#diffusers.IFImg2ImgSuperResolutionPipeline

the error message:

image

I already logging in to huggingface. It seems that there is a bug when mapping the model name? I never used IF-I-XL, I used IF-I-IF instead.

kanttouchthis commented 1 year ago

the repo was renamed from IF-I-IF to IF-I-XL changing

pipe = IFImg2ImgPipeline.from_pretrained(
    "DeepFloyd/IF-I-IF-v1.0",
    variant="fp16",
    torch_dtype=torch.float16,
)

to

pipe = IFImg2ImgPipeline.from_pretrained(
    "DeepFloyd/IF-I-XL-v1.0",
    variant="fp16",
    torch_dtype=torch.float16,
)

fixes this. this only needs to be done for stage 1