core.models.huggingface.clip_model

Module Contents

Classes

SiglipHuggingFaceModel

Wrapper for Siglip HuggingFace models.

API

class core.models.huggingface.clip_model.SiglipHuggingFaceModel(config)

Bases: megatron.core.models.huggingface.HuggingFaceModule

Wrapper for Siglip HuggingFace models.

Initialization

_fsdp_modules

None

forward(*args, **kwargs)

Siglip forward.