oops

huggingface · Oct 11, 2023 · 3466e30 · 3466e30
1 parent 3f4a279
commit 3466e30
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/trl/models/modeling_sd_base.py b/trl/models/modeling_sd_base.py
@@ -29,9 +29,9 @@
 
 
 if is_diffusers_greater_than("0.20.2"):
-    from diffusers.utils import randn_tensor
-else:
     from diffusers.utils.torch_utils import randn_tensor
+else:
+    from diffusers.utils import randn_tensor
 
 
 @dataclass