pengdaqian commited on
Commit
702657d
1 Parent(s): 852d36a
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -58,7 +58,7 @@ def infer(prompt: str, negative: str, width: int, height: int, sampler: str, ste
58
  if torch.cuda.is_available():
59
  generator = torch.Generator(device="cuda").manual_seed(seed)
60
  else:
61
- generator = torch.manual_seed(seed)
62
  if sampler == "EulerDiscrete":
63
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
64
  elif sampler == "EulerAncestralDiscrete":
 
58
  if torch.cuda.is_available():
59
  generator = torch.Generator(device="cuda").manual_seed(seed)
60
  else:
61
+ generator = None
62
  if sampler == "EulerDiscrete":
63
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
64
  elif sampler == "EulerAncestralDiscrete":