mfai.pytorch.models.llms¶

class mfai.pytorch.models.llms.FreezeMLMMixin[source]¶

A Mixin for (un)freezing llm and vision stages of a multimodal model.

freeze_llm()[source]¶

Freeze the LLM layers (not the vision layers).

freeze_vision()[source]¶

Freeze the vision encoder layers.

unfreeze_llm()[source]¶

Unfreeze the LLM layers.

unfreeze_vision()[source]¶

Unfreeze the vision encoder layers.