rahul7star commited on
Commit
aaf4035
·
verified ·
1 Parent(s): 6a66448

Update app_lora1.py

Browse files
Files changed (1) hide show
  1. app_lora1.py +1 -1
app_lora1.py CHANGED
@@ -76,7 +76,7 @@ def generate_image(prompt, height, width, steps, seed, guidance_scale):
76
  if pipe is None:
77
  return None, [], "❌ Pipeline not initialized"
78
 
79
- generator = torch.Generator(DEVICE).manual_seed(int(seed))
80
  placeholder = Image.new("RGB", (width, height), (255, 255, 255))
81
  previews = []
82
 
 
76
  if pipe is None:
77
  return None, [], "❌ Pipeline not initialized"
78
 
79
+ generator = torch.Generator().manual_seed(int(seed))
80
  placeholder = Image.new("RGB", (width, height), (255, 255, 255))
81
  previews = []
82