Spaces:
Runtime error
Runtime error
pengdaqian
commited on
Commit
•
702657d
1
Parent(s):
852d36a
fix more
Browse files
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 =
|
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":
|