Update app_demo.py
Browse files- app_demo.py +2 -2
app_demo.py
CHANGED
|
@@ -14,12 +14,12 @@ from safetensors.torch import load_file
|
|
| 14 |
from concurrent.futures import ThreadPoolExecutor
|
| 15 |
import uuid
|
| 16 |
#import cv2
|
| 17 |
-
model_id = "Lykon/dreamshaper-7" #"openskyml/lcm-lora-sdxl-turbo" #"SimianLuo/LCM_Dreamshaper_v7"
|
| 18 |
DESCRIPTION = '''# Fast Stable Diffusion CPU with Latent Consistency Model
|
| 19 |
Distilled from [Dreamshaper v7](https://huggingface.co/Lykon/dreamshaper-7) fine-tune of [Stable Diffusion v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5) with only 4,000 training iterations (~32 A100 GPU Hours). [Project page](https://latent-consistency-models.github.io)
|
| 20 |
'''
|
| 21 |
if not torch.cuda.is_available():
|
| 22 |
-
DESCRIPTION += "\n<
|
| 23 |
|
| 24 |
MAX_SEED = np.iinfo(np.int32).max
|
| 25 |
CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES") == "1"
|
|
|
|
| 14 |
from concurrent.futures import ThreadPoolExecutor
|
| 15 |
import uuid
|
| 16 |
#import cv2
|
| 17 |
+
model_id = "Lykon/dreamshaper-xl-v2-turbo" #"Lykon/dreamshaper-7" #"openskyml/lcm-lora-sdxl-turbo" #"SimianLuo/LCM_Dreamshaper_v7"
|
| 18 |
DESCRIPTION = '''# Fast Stable Diffusion CPU with Latent Consistency Model
|
| 19 |
Distilled from [Dreamshaper v7](https://huggingface.co/Lykon/dreamshaper-7) fine-tune of [Stable Diffusion v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5) with only 4,000 training iterations (~32 A100 GPU Hours). [Project page](https://latent-consistency-models.github.io)
|
| 20 |
'''
|
| 21 |
if not torch.cuda.is_available():
|
| 22 |
+
DESCRIPTION += "\n<a href="https://huggingface.co/spaces/charliebaby2023/Fast_Stable_diffusion_CPU/edit/main/app_demo.py">edit this file</a>"
|
| 23 |
|
| 24 |
MAX_SEED = np.iinfo(np.int32).max
|
| 25 |
CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES") == "1"
|