Update app.py

app.py

@@ -98,7 +98,9 @@ FTP_PASS = os.getenv("FTP_PASS")
 FTP_DIR = os.getenv("FTP_DIR")
 
 # os.putenv('TORCH_LINALG_PREFER_CUSOLVER','1')
+
 os.putenv('HF_HUB_ENABLE_HF_TRANSFER','1')
+
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 upscaler = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device("cuda:0"))
@@ -295,7 +297,7 @@ def generate_30(
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     seed = random.randint(0, MAX_SEED)
-    generator = torch.Generator(device='cuda').manual_seed(seed)
+    generator = torch.Generator(device='cpu').manual_seed(seed)
     options = {
         "prompt": [prompt],
         "negative_prompt": [negative_prompt],
@@ -346,7 +348,7 @@ def generate_60(
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     seed = random.randint(0, MAX_SEED)
-    generator = torch.Generator(device='cuda').manual_seed(seed)
+    generator = torch.Generator(device='cpu').manual_seed(seed)
     options = {
         "prompt": [prompt],
         "negative_prompt": [negative_prompt],
@@ -387,7 +389,7 @@ def generate_90(
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     seed = random.randint(0, MAX_SEED)
-    generator = torch.Generator(device='cuda').manual_seed(seed)
+    generator = torch.Generator(device='cpu').manual_seed(seed)
     options = {
         "prompt": [prompt],
         "negative_prompt": [negative_prompt],