wangshuai6 commited on
Commit
67d7248
·
1 Parent(s): 323e5b5
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -147,7 +147,7 @@ if __name__ == "__main__":
147
  pipeline = Pipeline(vae, denoiser, conditioner, diffusion_sampler, args.resolution, classlabels2id)
148
 
149
  with gr.Blocks() as demo:
150
- gr.Markdown("DDT")
151
  with gr.Row():
152
  with gr.Column(scale=1):
153
  num_steps = gr.Slider(minimum=1, maximum=100, step=1, label="num steps", value=50)
 
147
  pipeline = Pipeline(vae, denoiser, conditioner, diffusion_sampler, args.resolution, classlabels2id)
148
 
149
  with gr.Blocks() as demo:
150
+ gr.Markdown("DDT: Decoupled Diffusion Transformer on ImageNet 512x512")
151
  with gr.Row():
152
  with gr.Column(scale=1):
153
  num_steps = gr.Slider(minimum=1, maximum=100, step=1, label="num steps", value=50)
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  lightning==2.5.0.post0
2
  omegaconf==2.3.0
3
- numpy==1.26.4
4
  torch==2.2.0
5
  diffusers==0.30.0
6
  jsonargparse[signatures]>=4.27.7
 
1
  lightning==2.5.0.post0
2
  omegaconf==2.3.0
3
+ numpy==1.26.0
4
  torch==2.2.0
5
  diffusers==0.30.0
6
  jsonargparse[signatures]>=4.27.7