Update app.py
Browse files
app.py
CHANGED
@@ -31,12 +31,12 @@ def random_seed(seed: int, randomize: bool):
|
|
31 |
return seed
|
32 |
|
33 |
def generate(
|
|
|
34 |
seed: int = 0,
|
35 |
width: int = 1024,
|
36 |
height: int = 1024,
|
37 |
-
prompt: str = "a cat"
|
38 |
):
|
39 |
-
generator = torch.Generator().manual_seed(
|
40 |
|
41 |
return pipe(
|
42 |
prompt=prompt,
|
|
|
31 |
return seed
|
32 |
|
33 |
def generate(
|
34 |
+
prompt: str = "a cat",
|
35 |
seed: int = 0,
|
36 |
width: int = 1024,
|
37 |
height: int = 1024,
|
|
|
38 |
):
|
39 |
+
generator = torch.Generator().manual_seed(seed)
|
40 |
|
41 |
return pipe(
|
42 |
prompt=prompt,
|