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":
|