fix naming conflict, notify #49

Kahsolt · Feb 2, 2024 · 9c532f3 · 9c532f3
1 parent 7481190
commit 9c532f3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/scripts/prompt_travel.py b/scripts/prompt_travel.py
@@ -833,6 +833,7 @@ def ssim_loop(
         inter_neg_hidden,
     ):
         """Copied from shift-attentions plugin: https://github.com/yownas/shift-attention/blob/0129f6b99109f6f7c9e4e2bee0d1dc5f96e62506/scripts/shift_attention.py#L268"""
+        import torchvision.transforms as T
 
         dist_per_image = 1 / (len(prompt_images) - 1)
         dists = [dist_per_image * (i) for i, _ in enumerate(prompt_images)]