leowajda commited on
Commit
bde29c8
ยท
1 Parent(s): b7f0ee7

add more examples

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +7 -4
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- title: Diffusion Model
3
  emoji: ๐ŸŒผ๐ŸŒธ๐Ÿค—๐Ÿต๏ธ๐ŸŒบ
4
  colorFrom: yellow
5
  colorTo: red
 
1
  ---
2
+ title: Unconditional Image Generation Through Denoising Diffusion Implicit Models
3
  emoji: ๐ŸŒผ๐ŸŒธ๐Ÿค—๐Ÿต๏ธ๐ŸŒบ
4
  colorFrom: yellow
5
  colorTo: red
app.py CHANGED
@@ -4,7 +4,8 @@ import tensorflow as tf
4
  from huggingface_hub import from_pretrained_keras
5
  from diffusion_sampler import DiffusionSampler
6
 
7
- print('\n'.join([f'- {device.name}' for device in tf.config.list_physical_devices('GPU')]) or 'No GPU devices found.')
 
8
 
9
  scheduler_button = gr.Radio(
10
  choices=["Linear", "Cosine"],
@@ -124,9 +125,11 @@ demo = gr.Interface(
124
  cache_examples=True,
125
  title="""Unconditional Image Generation Through Denoising Diffusion Implicit Models""",
126
  examples=[
127
- ["Linear", "DDPM", "Linear", True, 1_000, 10],
128
- ["Cosine", "DDPM", "Linear", True, 750, 20],
129
- ["Linear", "DDIM", "Linear", True, 750, 20]
 
 
130
  ],
131
  description="""
132
  <p align="center">
 
4
  from huggingface_hub import from_pretrained_keras
5
  from diffusion_sampler import DiffusionSampler
6
 
7
+ devices = '\n'.join([f'- {device.name}' for device in tf.config.list_physical_devices('GPU')]) or 'No GPU devices found.'
8
+ print(f"GPUs available: {devices}")
9
 
10
  scheduler_button = gr.Radio(
11
  choices=["Linear", "Cosine"],
 
125
  cache_examples=True,
126
  title="""Unconditional Image Generation Through Denoising Diffusion Implicit Models""",
127
  examples=[
128
+ ["Linear", "DDPM", "Linear", True, 1_000, 25],
129
+ ["Linear", "DDPM", "Linear", False, 1_000, 25],
130
+ ["Cosine", "DDPM", "Linear", False, 750, 25],
131
+ ["Cosine", "DDPM", "Linear", True, 750, 25],
132
+ ["Linear", "DDIM", "Linear", True, 750, 25],
133
  ],
134
  description="""
135
  <p align="center">