Spaces:

laion
/

CoCa

Runtime error

gpucce commited on Feb 4, 2023

Commit

95c2912

1 Parent(s): 21a0d95

add hf hub

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ device = torch.device("cuda")
 model, _, transform = open_clip.create_model_and_transforms(
     "coca_ViT-L-14",
-    pretrained="mscoco_finetuned_laion2B-s13B-b90k"
 )
 model.to(device)

 model, _, transform = open_clip.create_model_and_transforms(
     "coca_ViT-L-14",
+    pretrained="hf-hub:laion2B-s13B-b90k-mscoco-2014.pt"
 )
 model.to(device)