update

src/pix2pix_turbo.py

@@ -109,7 +109,8 @@ class Pix2Pix_Turbo(torch.nn.Module):
         _sd_unet = unet.state_dict()
         for k in sd["state_dict_unet"]: _sd_unet[k] = sd["state_dict_unet"][k]
         unet.load_state_dict(_sd_unet)
-        unet.enable_xformers_memory_efficient_attention()
+        if device == "cuda":
+            unet.enable_xformers_memory_efficient_attention()
         _sd_vae = vae.state_dict()
         for k in sd["state_dict_vae"]: _sd_vae[k] = sd["state_dict_vae"][k]
         vae.load_state_dict(_sd_vae)