adding accelerate and testing CPU

app.py

@@ -37,8 +37,8 @@ class Demo:
 
         self.training = False
         self.generating = False
-        self.device = 'cuda'
-        self.weight_dtype = torch.float16
+        self.device = 'cpu'
+        self.weight_dtype = torch.float32
         self.pipe = StableDiffusionXLPipeline.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=self.weight_dtype).to(self.device)
 
         with gr.Blocks() as demo:

requirements.txt

@@ -20,3 +20,4 @@ torchvision==0.15.2
 tqdm==4.64.1
 transformers==4.27.4
 wandb==0.12.21
+accelerate==0.16.0