aimet
aimet copied to clipboard
import Error: cannot import name 'apply_adaround_te' from 'aimet_torch.quantsim'
I am using AIMET for the quantification of Stable-diffusion v1-5 according to the guidance in the following link
https://docs.qualcomm.com/bundle/publicresource/topics/80-64748-1/model_updates.html.
However, during the process of PTQ quantification, I encountered the following error:
ImportError: cannot import name 'apply_adaround_te' from 'aimet_torch.quantsim'
The AIMET version I am using is as follows:
aimet_torch torch-gpu-1.31.0
It is downloaded from the following link:
https://github.com/quic/aimet/releases/download/1.31.0/aimet_torch-torch_gpu_1.31.0-cp38-cp38-linux_x86_64.whl
I followed the instructions in the following file to build docker, and then installed the wheel package in it,
https://github.com/quic/aimet/blob/develop/Jenkins/Dockerfile.torch-gpu
Please help me point out what changes I should make to solve this problem, thank you very much!