Missing Package from requirements.txt

[Night1099]

default training script for pix2pix-turbo
Fails

Traceback (most recent call last):
File "/workspace/img2img-turbo/src/train_pix2pix_turbo.py", line 307, in
main(args)
File "/workspace/img2img-turbo/src/train_pix2pix_turbo.py", line 65, in main
import vision_aided_loss
ModuleNotFoundError: No module named 'vision_aided_loss'
Traceback (most recent call last):
File "/usr/local/bin/accelerate", line 8, in
sys.exit(main())
File "/usr/local/lib/python3.10/dist-packages/accelerate/commands/accelerate_cli.py", line 46, in main
args.func(args)
File "/usr/local/lib/python3.10/dist-packages/accelerate/commands/launch.py", line 1075, in launch_command
simple_launcher(args)
File "/usr/local/lib/python3.10/dist-packages/accelerate/commands/launch.py", line 681, in simple_launcher
raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)
subprocess.CalledProcessError: Command '['/usr/bin/python', 'train_pix2pix_turbo.py', '--pretrained_model_name_or_path=stabilityai/sd-turbo', '--output_dir=output/pix2pix_turbo/DiffuseToHeight', '--dataset_folder=data/DiffuseToHeight', '--resolution=512', '--train_batch_size=2', '--enable_xformers_memory_efficient_attention', '--viz_freq', '25', '--track_val_fid', '--report_to', 'wandb', '--tracker_project_name', 'DiffusetoHeight']' returned non-zero exit status 1.

[Night1099]

turned out to be problem when using torch dynamo

to fix

pip install vision_aided_loss

dont use torch dynamo