patrickvonplaten's picture
up
267e820
raw history blame
No virus
454 Bytes
#!/usr/bin/env python3
from transformers import CLIPTextModel
print("This should work!:")
model = CLIPTextModel.from_pretrained("./text_encoder", variant="no_ema")
print("This should work!:")
model = CLIPTextModel.from_pretrained("./text_encoder", variant="fp16")
print("This should work!:")
model = CLIPTextModel.from_pretrained("./text_encoder")
print("This should NOT work!:")
model = CLIPTextModel.from_pretrained("./text_encoder", variant="other")