modify device to cpu

stable_diffusion.py

@@ -18,7 +18,7 @@ from torch import Tensor
 from torch.nn import functional as F
 from torch.nn.parameter import Parameter
 
-device = "mps"
+device = "cpu"
 
 def apply_seq(seqs, x):
     for seq in seqs:
@@ -830,6 +830,7 @@ class Args(object):
         self.input_mask = input_mask
         self.input_image_strength = input_image_strength
         self.unphrase = unphrase
+        device = self.device
 
 from PIL import Image
 
@@ -999,6 +1000,7 @@ class Generate2img(Module):
 @lru_cache()
 def generate2img(phrase, steps, model_file, guidance_scale, img_width, img_height, seed, device, input_image, input_mask, input_image_strength=0.5, unphrase=""):
     try:
+
         args = Args(phrase, steps, None, guidance_scale, img_width, img_height, seed, device, model_file, input_image, input_mask, input_image_strength, unphrase)
         im = Generate2img.instance(args).forward(args.phrase)
         im = Generate2img.instance(args).decode_latent2img(im)