Open hiredd opened 2 months ago
change
# clip_model, _preprocess = clip.load("ViT-B/32", device=device) if opt['path']['daclip'] is not None: clip_model, preprocess = open_clip.create_model_from_pretrained('daclip_ViT-B-32', pretrained=opt['path']['daclip']) else: clip_model, _, preprocess = open_clip.create_model_and_transforms('ViT-B-32', pretrained='laion2b_s34b_b79k') tokenizer = open_clip.get_tokenizer('ViT-B-32') clip_model = clip_model.to(device)
to
# clip_model, _preprocess = clip.load("ViT-B/32", device=device) if opt['path']['daclip'] is not None: clip_model, preprocess = open_clip.create_model_from_pretrained('daclip_ViT-L-14', pretrained=opt['path']['daclip']) else: clip_model, _, preprocess = open_clip.create_model_and_transforms('ViT-B-32', pretrained='laion2b_s34b_b79k') tokenizer = open_clip.get_tokenizer('ViT-L-14') clip_model = clip_model.to(device)
Thanks!! I'll update it later!
change
to