Spaces:
Runtime error
Runtime error
Mateo Fidabel
commited on
Commit
·
b123ec9
1
Parent(s):
42d64c8
Removed the profiler
Browse files
app.py
CHANGED
|
@@ -4,7 +4,6 @@ from PIL import Image
|
|
| 4 |
from flax.jax_utils import replicate
|
| 5 |
from flax.training.common_utils import shard
|
| 6 |
from diffusers import FlaxControlNetModel, FlaxStableDiffusionControlNetPipeline
|
| 7 |
-
import jax.profiler
|
| 8 |
import jax.numpy as jnp
|
| 9 |
import numpy as np
|
| 10 |
import gc
|
|
@@ -22,8 +21,6 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
|
|
| 22 |
params["controlnet"] = controlnet_params
|
| 23 |
p_params = replicate(params)
|
| 24 |
|
| 25 |
-
jax.profiler.save_device_memory_profile("memory.prof")
|
| 26 |
-
|
| 27 |
# Description
|
| 28 |
title = "# 🧨 ControlNet on Segment Anything 🤗"
|
| 29 |
description = """This is a demo on 🧨 ControlNet based on Meta's [Segment Anything Model](https://segment-anything.com/).
|
|
|
|
| 4 |
from flax.jax_utils import replicate
|
| 5 |
from flax.training.common_utils import shard
|
| 6 |
from diffusers import FlaxControlNetModel, FlaxStableDiffusionControlNetPipeline
|
|
|
|
| 7 |
import jax.numpy as jnp
|
| 8 |
import numpy as np
|
| 9 |
import gc
|
|
|
|
| 21 |
params["controlnet"] = controlnet_params
|
| 22 |
p_params = replicate(params)
|
| 23 |
|
|
|
|
|
|
|
| 24 |
# Description
|
| 25 |
title = "# 🧨 ControlNet on Segment Anything 🤗"
|
| 26 |
description = """This is a demo on 🧨 ControlNet based on Meta's [Segment Anything Model](https://segment-anything.com/).
|