remove unused import

2022-06-16 18:07:59 +02:00 · 2022-06-16 18:07:59 +02:00 · cdf26c55f5
parent cc45831ec6
commit cdf26c55f5
1 changed files with 1 additions and 1 deletions
--- a/src/diffusers/models/unet_grad_tts.py
+++ b/src/diffusers/models/unet_grad_tts.py
@ -3,7 +3,7 @@ import math
 import torch

 try:
-    from einops import rearrange, repeat
+    from einops import rearrange
 except:
    print("Einops is not installed")
    pass