#!/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")