Spaces:
Running
on
Zero
Running
on
Zero
prithivMLmods
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -99,14 +99,14 @@ NUM_IMAGES_PER_PROMPT = 1
|
|
99 |
|
100 |
if torch.cuda.is_available():
|
101 |
pipe = DiffusionPipeline.from_pretrained(
|
102 |
-
"SG161222/
|
103 |
torch_dtype=torch.float16,
|
104 |
use_safetensors=True,
|
105 |
add_watermarker=False,
|
106 |
variant="fp16"
|
107 |
)
|
108 |
pipe2 = DiffusionPipeline.from_pretrained(
|
109 |
-
"SG161222/
|
110 |
torch_dtype=torch.float16,
|
111 |
use_safetensors=True,
|
112 |
add_watermarker=False,
|
|
|
99 |
|
100 |
if torch.cuda.is_available():
|
101 |
pipe = DiffusionPipeline.from_pretrained(
|
102 |
+
"SG161222/RealVisXL_V3.0_Turbo",
|
103 |
torch_dtype=torch.float16,
|
104 |
use_safetensors=True,
|
105 |
add_watermarker=False,
|
106 |
variant="fp16"
|
107 |
)
|
108 |
pipe2 = DiffusionPipeline.from_pretrained(
|
109 |
+
"SG161222/RealVisXL_V2.02_Turbo",
|
110 |
torch_dtype=torch.float16,
|
111 |
use_safetensors=True,
|
112 |
add_watermarker=False,
|