tokenflow

Runtime error

cocktailpeanut commited on Sep 5, 2023

Commit

21e2861

•

1 Parent(s): 9ed3831

cuda

Files changed (1) hide show

preprocess_utils.py CHANGED Viewed

@@ -84,7 +84,8 @@ class Preprocess(nn.Module):
             self.depth_maps = self.prepare_depth_maps()
         self.scheduler = scheduler
-        self.unet.enable_xformers_memory_efficient_attention()
         print(f'[INFO] loaded stable diffusion!')

             self.depth_maps = self.prepare_depth_maps()
         self.scheduler = scheduler
+        if self.device == "cuda":
+            self.unet.enable_xformers_memory_efficient_attention()
         print(f'[INFO] loaded stable diffusion!')