Update SAM AMG README with more descriptive install instructions#1337
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/1337
Note: Links to docs will display an error until the docs builds have been completed. ❌ 2 New FailuresAs of commit 3dcb54f with merge base ed76e9c ( NEW FAILURES - The following jobs have failed:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
| cd examples/sam2_amg_server/ | ||
| ``` | ||
|
|
||
| Download `sam2.1_hiera_large.pt` from https://github.com/facebookresearch/sam2?tab=readme-ov-file#download-checkpoints and put it into `~/checkpoints/sam2` |
There was a problem hiding this comment.
The download part I guess doesn't apply anymore now that the user can get this via the model type argument.
|
Looks good, let me know once you got a chance to give this another commit or two and then we can merge. |
| # Model | ||
| model: | ||
| _target_: sam2.modeling.sam2_base.SAM2Base | ||
| _target_: torchao._models.sam2.modeling.sam2_base.SAM2Base |
There was a problem hiding this comment.
Ah, these are the sam2 configs. There's sam2 and then there's sam2.1. I had to switched to sam2.1, but maybe sam2 is still useful? I updated the sam2.1 configs in b2e42ff#diff-12caf984b30e7028df7fb867a984c5ccad3fde0a0783df2af26db9cef860fbe5 , but in any case it doesn't hurt to have sam2 support as well.
…h#1337) * [AOTI] Remove the original model weights in Python deployment Summary: Fixes pytorch/torchchat#1302. Because AOTI-compiled model contains a copy of model weights, we need to release the corresponding eager model weights in the Python deployment path. * Revert "[AOTI] Remove the original model weights in Python deployment" This reverts commit 962ec0d913bbf6c30496560f12b4726445dce7da. * Refactor the code * Add setup_cache for aoti_package_path --------- Co-authored-by: Jack-Khuu <jack.khuu.7@gmail.com>
bhack
left a comment
There was a problem hiding this comment.
Can we recap in the Readme also the optimizations we have introduced with AO?
|
@bhack - Yes, we're working a more detailed description of this. |
@cpuhrsch
Installing was not so obvious, so this should help others trying to test and contribute.
There's some extra commits from amending, I'm still kinda garbage at git