aimet icon indicating copy to clipboard operation
aimet copied to clipboard

import Error: cannot import name 'apply_adaround_te' from 'aimet_torch.quantsim'

Open 1826133674 opened this issue 9 months ago • 0 comments

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!

1826133674 avatar Apr 30 '24 08:04 1826133674