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