Modelpublic

diffusers/tools

sourceHugging Facecreativeml-openrail-mupdated 3y agoView on Hugging Face

11likes27downloads

run_better_trans_clip.py15 linesDownload Raw Back to root

1#!/usr/bin/env python32from transformers import CLIPTextConfig, CLIPModel3import torch4 5config = CLIPTextConfig.from_pretrained("openMUSE/CLIP-ViT-L-14-DataComp.XL-s13B-b90K-penultimate")6 7model = CLIPModel.from_pretrained("laion/CLIP-ViT-L-14-DataComp.XL-s13B-b90K", text_config=config)8 9model.to_bettertransformer()10 11text_encoder = model.text_model12text_encoder = torch.compile(text_encoder)13 14 15