From cdf26c55f58e6dc552be7b2e44584630e7eab2f5 Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Thu, 16 Jun 2022 18:07:59 +0200 Subject: [PATCH] remove unused import --- src/diffusers/models/unet_grad_tts.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/unet_grad_tts.py b/src/diffusers/models/unet_grad_tts.py index 7bd6414b05..72009813ad 100644 --- 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