Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Fix TorchAO related bugs; revert device_map changes #10371

Merged
merged 23 commits into from
Dec 25, 2024
+350 −125
Merged

Fix TorchAO related bugs; revert device_map changes #10371

Changes from 1 commit
Commits
Show all changes
23 commits
Select commit Hold shift + click to select a range
03049aa
Revert "Add support for sharded models when TorchAO quantization is e…
a-r-r-o-w Dec 24, 2024
1a29a99
update tests
a-r-r-o-w Dec 24, 2024
c6651f9
udpate
a-r-r-o-w Dec 24, 2024
87bb2fe
update
a-r-r-o-w Dec 24, 2024
ba1269d
update
a-r-r-o-w Dec 24, 2024
bc47057
Merge branch 'main' into fix-torchao-related-bugs
a-r-r-o-w Dec 24, 2024
1873bb7
update device map tests
a-r-r-o-w Dec 24, 2024
d0b718a
apply review suggestions
a-r-r-o-w Dec 24, 2024
a10f19c
update
a-r-r-o-w Dec 24, 2024
fb8b44e
make style
a-r-r-o-w Dec 24, 2024
2bd9302
fix
a-r-r-o-w Dec 24, 2024
651666d
update docs
a-r-r-o-w Dec 24, 2024
d1b6405
update tests
a-r-r-o-w Dec 24, 2024
1dcd24e
update workflow
a-r-r-o-w Dec 24, 2024
e5dcdec
update
a-r-r-o-w Dec 24, 2024
1ae9fec
improve tests
a-r-r-o-w Dec 24, 2024
2663e94
allclose tolerance
a-r-r-o-w Dec 24, 2024
9f98833
Update src/diffusers/models/modeling_utils.py
a-r-r-o-w Dec 25, 2024
b7de749
Update tests/quantization/torchao/test_torchao.py
a-r-r-o-w Dec 25, 2024
77a3456
Merge branch 'main' into fix-torchao-related-bugs
a-r-r-o-w Dec 25, 2024
3e72979
improve tests
a-r-r-o-w Dec 25, 2024
a98a184
fix
a-r-r-o-w Dec 25, 2024
fa33949
update correct slices
a-r-r-o-w Dec 25, 2024
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
Prev Previous commit
Next Next commit
Loading
Loading status checks…
update workflow
@a-r-r-o-w
a-r-r-o-w committed Dec 24, 2024
commit 1dcd24ea1bcaaef378e9fb015ef0ab2f639ed8a5
2 changes: 2 additions & 0 deletions .github/workflows/nightly_tests.yml
View file
Original file line number Diff line number Diff line change
@@ -359,6 +359,8 @@ jobs:
test_location: "bnb"
- backend: "gguf"
test_location: "gguf"
- backend: "torchao"
test_location: "torchao"
runs-on:
group: aws-g6e-xlarge-plus
container: