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"
|