test

utils/tools.py

@@ -112,11 +112,11 @@ def get_pipeline(
 
     pipeline.scheduler = UniPCMultistepScheduler.from_config(pipeline.scheduler.config)
     pipeline.set_progress_bar_config()
-    pipeline = pipeline.to(device, dtype=torch.float16)
+    pipeline = pipeline.to(device, dtype=torch.float16 if torch.cuda.is_available() else torch.float32)
 
     if lora_path is not None:
         pipeline.load_lora_weights(lora_path)
-    if enable_xformers_memory_efficient_attention:
+    if enable_xformers_memory_efficient_attention and torch.cuda.is_available():
         pipeline.enable_xformers_memory_efficient_attention()
 
     return pipeline