apple
/

aimv2-large-patch14-224-lit

Zero-Shot Image Classification

feature-extraction

Model card Files Files and versions

michalk8 commited on Nov 28, 2024

Commit

1943b4b

·

1 Parent(s): ae9d0c0

Fix importw

Files changed (1) hide show

modeling_aimv2.py +1 -1

modeling_aimv2.py CHANGED Viewed

@@ -7,7 +7,7 @@ import math
 from transformers.modeling_attn_mask_utils import AttentionMaskConverter
 from transformers.utils import ModelOutput
-from configuration_aimv2 import AIMv2Config, AIMv2VisionConfig, AIMv2TextConfig
 from torch import nn
 from torch.nn import functional as F
 from transformers.modeling_outputs import BaseModelOutputWithNoAttention

 from transformers.modeling_attn_mask_utils import AttentionMaskConverter
 from transformers.utils import ModelOutput
+from .configuration_aimv2 import AIMv2Config, AIMv2VisionConfig, AIMv2TextConfig
 from torch import nn
 from torch.nn import functional as F
 from transformers.modeling_outputs import BaseModelOutputWithNoAttention