Commit 1efa9c8
authored
fix(deps): align torchao constraint in requirements.txt with pyproject.toml (#1298)
requirements.txt declares `torchao` unpinned while pyproject.toml declares
`torchao>=0.16.0,<0.17.0` for non-aarch64. A fresh
`pip install -r requirements.txt` on Windows therefore resolves torchao 0.18.0,
while the same file pins torch==2.7.1+cu128 for win32.
torchao 0.18.x imports `ScalingType` from `torch.nn.functional`, which does not
exist in torch 2.7.1, so loading the DiT checkpoint fails with:
ImportError: cannot import name 'ScalingType' from 'torch.nn.functional'
RuntimeError: Failed to load model with attention implementations
['sdpa', 'eager']
Installing the range already declared in pyproject.toml (torchao 0.16.0)
restores loading with no other change. Verified in a clean virtual environment
on Python 3.12.10: torchao resolves to 0.16.0, `import torchao.quantization`
succeeds, the released checkpoint loads on CUDA, and seeded generation produces
decoded audio identical to the same seeded request in a separately built
environment.
Mirrors the pyproject.toml markers exactly so the two declarations cannot drift
apart again.1 parent 55476ee commit 1efa9c8
1 file changed
Lines changed: 6 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
34 | 34 | | |
35 | 35 | | |
36 | 36 | | |
37 | | - | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
| 41 | + | |
| 42 | + | |
38 | 43 | | |
39 | 44 | | |
40 | 45 | | |
| |||
0 commit comments