Spaces:
Runtime error
Runtime error
Commit
·
2383edf
1
Parent(s):
8c37893
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ from diffusers import (
|
|
38 |
UniPCMultistepScheduler,
|
39 |
)
|
40 |
|
41 |
-
DESCRIPTION = "
|
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", "
|
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
|