Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

redirect flamingo import into llama3.2 to solve import issue #1206

Closed
wants to merge 1 commit into from
Closed

redirect flamingo import into llama3.2 to solve import issue #1206

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions torchchat/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@
)
from torch.nn import functional as F

from torchtune.models.flamingo import flamingo_decoder, flamingo_vision_encoder
from torchtune.models.llama3_2_vision import llama3_2_vision_decoder, llama3_2_vision_encoder
from torchtune.models.llama3_1._component_builders import llama3_1 as llama3_1_builder
from torchtune.modules.model_fusion import DeepFusionModel
from torchtune.models.clip import clip_vision_encoder
Expand Down Expand Up @@ -213,7 +213,7 @@ def _llama3_1(cls):
def _flamingo(cls):
return cls(
model_type=ModelType.Flamingo,
modules={"encoder": flamingo_vision_encoder, "decoder": flamingo_decoder},
modules={"encoder": llama3_2_vision_encoder, "decoder": llama3_2_vision_decoder},
fusion_class=DeepFusionModel,
)

Expand Down
Loading