Skip to content
Merged
Show file tree
Hide file tree
Changes from 6 commits
Commits
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
3 changes: 3 additions & 0 deletions examples/models/llama3_2_vision/preprocess/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,9 @@ class PreprocessConfig:
max_num_tiles: int = 4
tile_size: int = 224
antialias: bool = False
# Used for reference eager model from torchtune.
resize_to_max_canvas: bool = False
possible_resolutions: Optional[List[Tuple[int, int]]] = None


class CLIPImageTransformModel(EagerModelBase):
Expand Down
Loading
Loading