Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
4bacf14
1
Parent(s):
9f53447
Update app.py
Browse files
app.py
CHANGED
@@ -87,6 +87,7 @@ def generate(
|
|
87 |
decoder_pipeline.to(device)
|
88 |
|
89 |
generator = torch.Generator().manual_seed(seed)
|
|
|
90 |
prior_output = prior_pipeline(
|
91 |
prompt=prompt,
|
92 |
height=height,
|
|
|
87 |
decoder_pipeline.to(device)
|
88 |
|
89 |
generator = torch.Generator().manual_seed(seed)
|
90 |
+
print("prior_num_inference_steps: ", prior_num_inference_steps)
|
91 |
prior_output = prior_pipeline(
|
92 |
prompt=prompt,
|
93 |
height=height,
|