diff --git a/src/diffusers/models/transformers/transformer_kandinsky.py b/src/diffusers/models/transformers/transformer_kandinsky.py
index 6dec8d93ac..24b2c4ae99 100644
--- a/src/diffusers/models/transformers/transformer_kandinsky.py
+++ b/src/diffusers/models/transformers/transformer_kandinsky.py
@@ -15,7 +15,6 @@
 import math
 from typing import Any, Dict, List, Optional, Tuple, Union
 
-from einops import rearrange
 import torch
 import torch.nn as nn
 import torch.nn.functional as F