Spaces:
Paused
Paused
add more examples
Browse files
README.md
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
---
|
| 2 |
-
title: Diffusion
|
| 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 |
-
|
|
|
|
| 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,
|
| 128 |
-
["
|
| 129 |
-
["
|
|
|
|
|
|
|
| 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">
|