Update onnx/image_embedding_phi3_v_for_onnx.py
Browse files
onnx/image_embedding_phi3_v_for_onnx.py
CHANGED
|
@@ -39,7 +39,8 @@ CLIP_VIT_LARGE_PATCH14_336_CONFIG = CLIPVisionConfig(
|
|
| 39 |
num_channels=3,
|
| 40 |
num_hidden_layers=24,
|
| 41 |
patch_size=14,
|
| 42 |
-
projection_dim=768
|
|
|
|
| 43 |
)
|
| 44 |
|
| 45 |
class Phi3ImageEmbedding(nn.Module):
|
|
|
|
| 39 |
num_channels=3,
|
| 40 |
num_hidden_layers=24,
|
| 41 |
patch_size=14,
|
| 42 |
+
projection_dim=768,
|
| 43 |
+
attn_implementation="eager"
|
| 44 |
)
|
| 45 |
|
| 46 |
class Phi3ImageEmbedding(nn.Module):
|