search

sayakpaul / diffusers-torchao

End-to-end recipes for optimizing diffusion models with torchao and diffusers (inference and FP8 training).
Apache License 2.0
274 stars 8 forks source link

Share requirements.txt file for this repo #24

Closed Tushar-ml closed 2 months ago

Tushar-ml commented 2 months ago

Hi @sayakpaul , could you please share requirements file, as they are some dependency issues, like this 'FakeTensor' object has no attribute '_quantized_linear_op'

sayakpaul commented 2 months ago

You should be able to use it with torch nightly and torchao installed from the source. Since these change frequently, didn't think to make a requirements.txt. If the import problem is stemming from torch or torchao, you should open an issue on those repositories.

For other dependencies, 

pip install diffusers transformers accelerate tabulate sentencepiece

Sorry for the inconvenience.