Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ def check_text(prompt, negative=""):
|
|
25 |
return True
|
26 |
return False
|
27 |
|
28 |
-
DESCRIPTION = """
|
29 |
if not torch.cuda.is_available():
|
30 |
DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
|
31 |
|
@@ -41,7 +41,8 @@ NUM_IMAGES_PER_PROMPT = 1
|
|
41 |
|
42 |
if torch.cuda.is_available():
|
43 |
pipe = DiffusionPipeline.from_pretrained(
|
44 |
-
"
|
|
|
45 |
torch_dtype=torch.float16,
|
46 |
use_safetensors=True,
|
47 |
add_watermarker=False,
|
|
|
25 |
return True
|
26 |
return False
|
27 |
|
28 |
+
DESCRIPTION = """"""
|
29 |
if not torch.cuda.is_available():
|
30 |
DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
|
31 |
|
|
|
41 |
|
42 |
if torch.cuda.is_available():
|
43 |
pipe = DiffusionPipeline.from_pretrained(
|
44 |
+
"stabilityai/sdxl-turbo",
|
45 |
+
##"SG161222/RealVisXL_V3.0",
|
46 |
torch_dtype=torch.float16,
|
47 |
use_safetensors=True,
|
48 |
add_watermarker=False,
|