multimodalart HF staff commited on
Commit
4bacf14
1 Parent(s): 9f53447

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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,