Update app.py

app.py

@@ -13,9 +13,10 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 import torch
-from diffusers import AutoencoderKL, StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
+from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
+#from diffusers import AutoencoderKL
 from typing import Tuple
-from transformers import AutoTokenizer, AutoModelForCausalLM
+#from transformers import AutoTokenizer, AutoModelForCausalLM
 import paramiko
 
 #os.system("chmod +x ./cusparselt.sh")
@@ -182,7 +183,7 @@ def generate(
     global models
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator(device='cuda').manual_seed(seed)
+    generator = torch.Generator(device='cpu').manual_seed(seed)
 
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)