Bootloader commited on
Commit
2383edf
·
1 Parent(s): 8c37893

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,7 +38,7 @@ from diffusers import (
38
  UniPCMultistepScheduler,
39
  )
40
 
41
- DESCRIPTION = "Animagine XL 2.0"
42
 
43
  if not torch.cuda.is_available():
44
  DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU. </p>"
@@ -52,7 +52,7 @@ MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "2048"))
52
  USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
53
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
54
 
55
- MODEL = os.getenv("MODEL", "Linaqruf/animagine-xl-2.0")
56
 
57
  torch.backends.cudnn.deterministic = True
58
  torch.backends.cudnn.benchmark = False
 
38
  UniPCMultistepScheduler,
39
  )
40
 
41
+ DESCRIPTION = "reimu xl"
42
 
43
  if not torch.cuda.is_available():
44
  DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU. </p>"
 
52
  USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
53
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
54
 
55
+ MODEL = os.getenv("MODEL", "Enferlain/218xl")
56
 
57
  torch.backends.cudnn.deterministic = True
58
  torch.backends.cudnn.benchmark = False