14 lines
361 B
Python
14 lines
361 B
Python
from transformers import CLIPVisionModel
|
|
|
|
from .._mixin import OstrisTransformersMixin
|
|
|
|
|
|
class CLIPVisionEncoder(CLIPVisionModel, OstrisTransformersMixin):
|
|
"""CLIP vision tower (wan21 i2v image conditioning)."""
|
|
|
|
aitk_subfolder = "image_encoder"
|
|
|
|
@classmethod
|
|
def get_transformer_block_names(cls):
|
|
return ["vision_model.encoder.layers"]
|