AdamOswald1 commited on
Commit
8fc010c
1 Parent(s): 233676c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -141,8 +141,8 @@ def inference(model_name, prompt, guidance, steps, n_images=1, width=512, height
141
 
142
  if torch.cuda.is_available():
143
  generator = torch.Generator('cuda').manual_seed(seed)
144
- else:
145
- generator = torch.Generator().manual_seed(seed)
146
 
147
  try:
148
  if img is not None:
 
141
 
142
  if torch.cuda.is_available():
143
  generator = torch.Generator('cuda').manual_seed(seed)
144
+ else:
145
+ generator = torch.Generator().manual_seed(seed)
146
 
147
  try:
148
  if img is not None: