showee commited on
Commit
af64cf3
·
1 Parent(s): 12fb4e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -114,7 +114,7 @@ def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None,
114
  if torch.cuda.is_available():
115
  generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
116
  else:
117
- generator = torch.Generator.manual_seed(seed) if seed != 0 else None
118
  prompt = f"{prefix} {prompt}" if auto_prefix else prompt
119
 
120
  try:
 
114
  if torch.cuda.is_available():
115
  generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
116
  else:
117
+ generator = torch.Generator().manual_seed(seed) if seed != 0 else None
118
  prompt = f"{prefix} {prompt}" if auto_prefix else prompt
119
 
120
  try: