Spaces:

orpatashnik
/

NestedAttentionEncoder

Sleeping

orpatashnik commited on Aug 4

Commit

ad42387

1 Parent(s): 38274da

small change

Files changed (1) hide show

nested_attention_pipeline.py CHANGED Viewed

@@ -62,7 +62,7 @@ class NestedAdapterInference:
         # load image encoder
         self.image_encoder = CLIPVisionModelWithProjection.from_pretrained(
-            self.image_encoder_path
         ).to(self.device, dtype=torch.float16)
         self.clip_image_processor = CLIPImageProcessor()

         # load image encoder
         self.image_encoder = CLIPVisionModelWithProjection.from_pretrained(
+            self.image_encoder_path, use_safetensors=True
         ).to(self.device, dtype=torch.float16)
         self.clip_image_processor = CLIPImageProcessor()