Update app.py

app.py

@@ -4,6 +4,8 @@ import torch
 from PIL import Image
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
+import os
+os.environ['CUDA_LAUNCH_BLOCKING'] = 1
 
 
 def convert_safetensors_to_bin(pipeline, state_dict, alpha = 0.4):
@@ -109,10 +111,7 @@ def error_str(error, title="Error"):
             {error}"""  if error else ""
 
 def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None, strength=0.5, neg_prompt="", auto_prefix=False):
-  if torch.cuda.is_available():
-    generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
-  else:
-    generator = torch.Generator.manual_seed(seed) if seed != 0 else None
+  generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
   prompt = f"{prefix} {prompt}" if auto_prefix else prompt
     
   try: