Spaces:
Running
on
Zero
Running
on
Zero
feat: add examples
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- app.py +3 -3
- example_outputs/.DS_Store +3 -0
- example_outputs/cat/.DS_Store +3 -0
- example_outputs/cat/bear/flux_iterations=0.png +3 -0
- example_outputs/cat/bear/flux_iterations=1.png +3 -0
- example_outputs/cat/bear/flux_iterations=2.png +3 -0
- example_outputs/cat/bear/flux_iterations=3.png +3 -0
- example_outputs/cat/bear/flux_iterations=4.png +3 -0
- example_outputs/cat/bear/flux_iterations=5.png +3 -0
- example_outputs/cat/bear/flux_iterations=6.png +3 -0
- example_outputs/cat/bear/flux_iterations=7.png +3 -0
- example_outputs/cat/puppy/sd3_iterations=0.png +3 -0
- example_outputs/cat/puppy/sd3_iterations=1.png +3 -0
- example_outputs/cat/puppy/sd3_iterations=2.png +3 -0
- example_outputs/cat/puppy/sd3_iterations=3.png +3 -0
- example_outputs/cat/puppy/sd3_iterations=4.png +3 -0
- example_outputs/cat/puppy/sd3_iterations=5.png +3 -0
- example_outputs/cat/puppy/sd3_iterations=6.png +3 -0
- example_outputs/cat_fridge/origami/sd3_iterations=0.png +3 -0
- example_outputs/cat_fridge/origami/sd3_iterations=1.png +3 -0
- example_outputs/cat_fridge/origami/sd3_iterations=2.png +3 -0
- example_outputs/cat_fridge/origami/sd3_iterations=3.png +3 -0
- example_outputs/cat_fridge/origami/sd3_iterations=4.png +3 -0
- example_outputs/cat_fridge/origami/sd3_iterations=5.png +3 -0
- example_outputs/cat_fridge/origami/sd3_iterations=6.png +3 -0
- example_outputs/cat_fridge/origami/sd3_iterations=7.png +3 -0
- example_outputs/cat_fridge/origami/sd3_iterations=8.png +3 -0
- example_outputs/corgi_walking/guinea_pig/flux_iterations=0.png +3 -0
- example_outputs/corgi_walking/guinea_pig/flux_iterations=1.png +3 -0
- example_outputs/corgi_walking/guinea_pig/flux_iterations=2.png +3 -0
- example_outputs/corgi_walking/guinea_pig/flux_iterations=3.png +3 -0
- example_outputs/corgi_walking/guinea_pig/flux_iterations=4.png +3 -0
- example_outputs/corgi_walking/guinea_pig/flux_iterations=5.png +3 -0
- example_outputs/corgi_walking/guinea_pig/flux_iterations=6.png +3 -0
- example_outputs/corgi_walking/guinea_pig/flux_iterations=7.png +3 -0
- example_outputs/corgi_walking/rabbit/sd3_iterations=0.png +3 -0
- example_outputs/corgi_walking/rabbit/sd3_iterations=1.png +3 -0
- example_outputs/corgi_walking/rabbit/sd3_iterations=2.png +3 -0
- example_outputs/corgi_walking/rabbit/sd3_iterations=3.png +3 -0
- example_outputs/corgi_walking/rabbit/sd3_iterations=4.png +3 -0
- example_outputs/corgi_walking/rabbit/sd3_iterations=5.png +3 -0
- example_outputs/corgi_walking/rabbit/sd3_iterations=6.png +3 -0
- example_outputs/corgi_walking/rabbit/sd3_iterations=7.png +3 -0
- example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=0.png +3 -0
- example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=1.png +3 -0
- example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=2.png +3 -0
- example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=3.png +3 -0
- example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=4.png +3 -0
- example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=5.png +3 -0
- example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=6.png +3 -0
app.py
CHANGED
|
@@ -14,7 +14,7 @@ from utils.sd3 import sd3_editing
|
|
| 14 |
|
| 15 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 16 |
|
| 17 |
-
|
| 18 |
pipe_flux = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
|
| 19 |
|
| 20 |
|
|
@@ -88,8 +88,8 @@ def FlowOpt_run(
|
|
| 88 |
|
| 89 |
if model_type_val == 'FLUX':
|
| 90 |
pipe = pipe_flux.to(device)
|
| 91 |
-
|
| 92 |
-
|
| 93 |
else:
|
| 94 |
raise NotImplementedError(f"Model type {model_type_val} not implemented")
|
| 95 |
|
|
|
|
| 14 |
|
| 15 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 16 |
|
| 17 |
+
pipe_sd3 = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
|
| 18 |
pipe_flux = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
|
| 19 |
|
| 20 |
|
|
|
|
| 88 |
|
| 89 |
if model_type_val == 'FLUX':
|
| 90 |
pipe = pipe_flux.to(device)
|
| 91 |
+
elif model_type_val == 'SD3':
|
| 92 |
+
pipe = pipe_sd3.to(device)
|
| 93 |
else:
|
| 94 |
raise NotImplementedError(f"Model type {model_type_val} not implemented")
|
| 95 |
|
example_outputs/.DS_Store
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:591a9ab3cc4d59aa617132303385bca06b806efd04843ab683191d87c1bdb752
|
| 3 |
+
size 6148
|
example_outputs/cat/.DS_Store
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3f6bf8c1be6414f8314c996ddbc0115263d3f60fb53f524aacf8277dc9f3c23f
|
| 3 |
+
size 6148
|
example_outputs/cat/bear/flux_iterations=0.png
ADDED
|
Git LFS Details
|
example_outputs/cat/bear/flux_iterations=1.png
ADDED
|
Git LFS Details
|
example_outputs/cat/bear/flux_iterations=2.png
ADDED
|
Git LFS Details
|
example_outputs/cat/bear/flux_iterations=3.png
ADDED
|
Git LFS Details
|
example_outputs/cat/bear/flux_iterations=4.png
ADDED
|
Git LFS Details
|
example_outputs/cat/bear/flux_iterations=5.png
ADDED
|
Git LFS Details
|
example_outputs/cat/bear/flux_iterations=6.png
ADDED
|
Git LFS Details
|
example_outputs/cat/bear/flux_iterations=7.png
ADDED
|
Git LFS Details
|
example_outputs/cat/puppy/sd3_iterations=0.png
ADDED
|
Git LFS Details
|
example_outputs/cat/puppy/sd3_iterations=1.png
ADDED
|
Git LFS Details
|
example_outputs/cat/puppy/sd3_iterations=2.png
ADDED
|
Git LFS Details
|
example_outputs/cat/puppy/sd3_iterations=3.png
ADDED
|
Git LFS Details
|
example_outputs/cat/puppy/sd3_iterations=4.png
ADDED
|
Git LFS Details
|
example_outputs/cat/puppy/sd3_iterations=5.png
ADDED
|
Git LFS Details
|
example_outputs/cat/puppy/sd3_iterations=6.png
ADDED
|
Git LFS Details
|
example_outputs/cat_fridge/origami/sd3_iterations=0.png
ADDED
|
Git LFS Details
|
example_outputs/cat_fridge/origami/sd3_iterations=1.png
ADDED
|
Git LFS Details
|
example_outputs/cat_fridge/origami/sd3_iterations=2.png
ADDED
|
Git LFS Details
|
example_outputs/cat_fridge/origami/sd3_iterations=3.png
ADDED
|
Git LFS Details
|
example_outputs/cat_fridge/origami/sd3_iterations=4.png
ADDED
|
Git LFS Details
|
example_outputs/cat_fridge/origami/sd3_iterations=5.png
ADDED
|
Git LFS Details
|
example_outputs/cat_fridge/origami/sd3_iterations=6.png
ADDED
|
Git LFS Details
|
example_outputs/cat_fridge/origami/sd3_iterations=7.png
ADDED
|
Git LFS Details
|
example_outputs/cat_fridge/origami/sd3_iterations=8.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/guinea_pig/flux_iterations=0.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/guinea_pig/flux_iterations=1.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/guinea_pig/flux_iterations=2.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/guinea_pig/flux_iterations=3.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/guinea_pig/flux_iterations=4.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/guinea_pig/flux_iterations=5.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/guinea_pig/flux_iterations=6.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/guinea_pig/flux_iterations=7.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/rabbit/sd3_iterations=0.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/rabbit/sd3_iterations=1.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/rabbit/sd3_iterations=2.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/rabbit/sd3_iterations=3.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/rabbit/sd3_iterations=4.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/rabbit/sd3_iterations=5.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/rabbit/sd3_iterations=6.png
ADDED
|
Git LFS Details
|
example_outputs/corgi_walking/rabbit/sd3_iterations=7.png
ADDED
|
Git LFS Details
|
example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=0.png
ADDED
|
Git LFS Details
|
example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=1.png
ADDED
|
Git LFS Details
|
example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=2.png
ADDED
|
Git LFS Details
|
example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=3.png
ADDED
|
Git LFS Details
|
example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=4.png
ADDED
|
Git LFS Details
|
example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=5.png
ADDED
|
Git LFS Details
|
example_outputs/cow_grass2/colorful_toy_bricks/flux_iterations=6.png
ADDED
|
Git LFS Details
|