use @clip-anytorch , thanks to @rom1504

This commit is contained in:
Phil Wang
2022-04-30 06:40:54 -07:00
parent 0d1c07c803
commit e2f9615afa
3 changed files with 4 additions and 10 deletions

View File

@@ -172,11 +172,7 @@ class OpenAIClipAdapter(BaseClipAdapter):
self,
name = 'ViT-B/32'
):
try:
import clip
except ImportError:
print('you must install openai clip in order to use this adapter - `pip install git+https://github.com/openai/CLIP.git` - more instructions at https://github.com/openai/CLIP#usage')
import clip
openai_clip, _ = clip.load(name)
super().__init__(openai_clip)
@@ -1636,4 +1632,3 @@ class DALLE2(nn.Module):
return images[0]
return images