charliebaby2023 commited on
Commit
8102d43
·
verified ·
1 Parent(s): 024c79f

Update app_demo.py

Browse files
Files changed (1) hide show
  1. 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<p>Running on CPU 🤩 This demo works on CPU 👌.</p>"
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"