Spaces:
Running
on
Zero
Running
on
Zero
Change to SD2-Community mirror due to SD2.1 depracation
Browse files
app.py
CHANGED
|
@@ -16,8 +16,8 @@ if os.getenv("SPACES_ZERO_GPU") == "true":
|
|
| 16 |
os.environ["SPACES_ZERO_GPU"] = "1"
|
| 17 |
|
| 18 |
|
| 19 |
-
avail_models = {'512x512': load_model('
|
| 20 |
-
'768x768': load_model('
|
| 21 |
}
|
| 22 |
|
| 23 |
compression_func = partial(latent_DDCM_compression.main, avail_models=avail_models)
|
|
@@ -88,7 +88,8 @@ DDCM can easily be utilized for perceptual image compression, as well as for sol
|
|
| 88 |
</br></br>
|
| 89 |
The tabs below correspond to demos of different practical applications. Open each tab to see the application's specific instructions.
|
| 90 |
</br></br>
|
| 91 |
-
<b>Note: The demos below rely on relatively old pre-trained diffusion models such as Stable Diffusion 2.1
|
|
|
|
| 92 |
"""
|
| 93 |
|
| 94 |
article = r"""
|
|
|
|
| 16 |
os.environ["SPACES_ZERO_GPU"] = "1"
|
| 17 |
|
| 18 |
|
| 19 |
+
avail_models = {'512x512': load_model('sd2-community/stable-diffusion-2-1-base', 1000, float16=True, device=torch.device("cpu"), compile=False)[0],
|
| 20 |
+
'768x768': load_model('sd2-community/stable-diffusion-2-1', 1000, float16=True, device=torch.device("cpu"), compile=False)[0]
|
| 21 |
}
|
| 22 |
|
| 23 |
compression_func = partial(latent_DDCM_compression.main, avail_models=avail_models)
|
|
|
|
| 88 |
</br></br>
|
| 89 |
The tabs below correspond to demos of different practical applications. Open each tab to see the application's specific instructions.
|
| 90 |
</br></br>
|
| 91 |
+
<b>Note: The demos below rely on relatively old pre-trained diffusion models such as Stable Diffusion 2.1
|
| 92 |
+
(Mirrored by <a href="https://huggingface.co/sd2-community">sd2-community</a>), simply for the purpose of demonstrating the capabilities of DDCM. Feel free to implement our DDCM-based methods using newer diffusion models to further improve performance.</b>
|
| 93 |
"""
|
| 94 |
|
| 95 |
article = r"""
|