Skip to content

Commit 71d5b3f

Browse files
committed
Add preprocess to ci
ghstack-source-id: 9f58d21 Pull Request resolved: #6544
1 parent e8d6118 commit 71d5b3f

File tree

3 files changed

+173
-131
lines changed

3 files changed

+173
-131
lines changed

examples/models/llama3_2_vision/preprocess/model.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@ class PreprocessConfig:
2626
max_num_tiles: int = 4
2727
tile_size: int = 224
2828
antialias: bool = False
29+
# Used for reference eager model from torchtune.
30+
resize_to_max_canvas: bool = False
31+
possible_resolutions: Optional[List[Tuple[int, int]]] = None
2932

3033

3134
class CLIPImageTransformModel(EagerModelBase):

0 commit comments

Comments
 (0)