alembics / disco-diffusion

Other
7.46k stars 1.13k forks source link

Cannot Use New OpenCLIP LAION Models #120

Closed 6630507 closed 2 years ago

6630507 commented 2 years ago

So far unable to use the new LAION CLIP models.

Disco crashes on step two with this error:

AttributeError                            Traceback (most recent call last)
Input In [19], in <cell line: 177>()
    175 if RN50x64: clip_models.append(clip.load('RN50x64', jit=False)[0].eval().requires_grad_(False).to(device))
    176 if RN101: clip_models.append(clip.load('RN101', jit=False)[0].eval().requires_grad_(False).to(device))
--> 177 if ViTB32_laion2b_e16: clip_models.append(open_clip.create_model('ViT-B-32', pretrained='laion2b_e16').eval().requires_grad_(False).to(device))
    178 if ViTB32_laion400m_e31: clip_models.append(open_clip.create_model('ViT-B-32', pretrained='laion400m_e31').eval().requires_grad_(False).to(device))
    179 if ViTB32_laion400m_32: clip_models.append(open_clip.create_model('ViT-B-32', pretrained='laion400m_e32').eval().requires_grad_(False).to(device))

**AttributeError: module 'open_clip' has no attribute 'create_model'**

Running Ubuntu 20.04, rebuilt by conda environment, and re-cloned everything.

Thanks for the fascinating software!

yizhangliu commented 2 years ago

Maybe you should pip install open-clip-torch==1.3.0.

6630507 commented 2 years ago

That did it, thanks!