Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ if torch.cuda.is_available():
|
|
13 |
else:
|
14 |
torch_dtype = torch.float32
|
15 |
|
16 |
-
|
17 |
pipe = pipe.to(device)
|
18 |
|
19 |
MAX_SEED = np.iinfo(np.int32).max
|
|
|
13 |
else:
|
14 |
torch_dtype = torch.float32
|
15 |
|
16 |
+
pipeline = DiffusionPipeline.from_pretrained("John6666/wai-ani-nsfw-ponyxl-v8-sdxl")
|
17 |
pipe = pipe.to(device)
|
18 |
|
19 |
MAX_SEED = np.iinfo(np.int32).max
|