running error: ModuleNotFoundError: No module named 'pretrainedmodels'

[LinyeLyuNeo]

Hi Chen, great work for adversarial attack using diffusion models, I am trying to run your code but getting the following errors:

python main.py --model_name "inception" --save_dir output --images_root demo/images --label_path demo/labels.txt Traceback (most recent call last): File "main.py", line 4, in import diff_latent_attack File "/home/Neo2/projects/DiffAttack/diff_latent_attack.py", line 8, in import other_attacks File "/home/Neo2/projects/DiffAttack/other_attacks.py", line 16, in from Finegrained_model import model as otherModel File "/home/Neo2/projects/DiffAttack/Finegrained_model/model.py", line 3, in from . import LoadModel File "/home/Neo2/projects/DiffAttack/Finegrained_model/LoadModel.py", line 4, in import pretrainedmodels ModuleNotFoundError: No module named 'pretrainedmodels'

I check the code file in LoadModel.py and find the error location: from torch import nn import torch from torchvision import models import pretrainedmodels from .config import pretrained_model

can you give me a hint on how to resolve this error so that I can run the attack on the demo image examples? Thanks!