Spaces:
Running
on
Zero
Running
on
Zero
Upload genimage.py
Browse files- genimage.py +3 -3
genimage.py
CHANGED
@@ -6,7 +6,7 @@ def load_pipeline():
|
|
6 |
import torch
|
7 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
8 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
9 |
-
"John6666/
|
10 |
torch_dtype=torch.float16,
|
11 |
)
|
12 |
pipe.to(device)
|
@@ -43,8 +43,8 @@ def generate_image(prompt, neg_prompt):
|
|
43 |
}
|
44 |
try:
|
45 |
images = pipe(
|
46 |
-
prompt=prompt + ",
|
47 |
-
negative_prompt=neg_prompt + ",
|
48 |
width=1024,
|
49 |
height=1024,
|
50 |
guidance_scale=7.5,
|
|
|
6 |
import torch
|
7 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
8 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
9 |
+
"John6666/t-ponynai3-v6-sdxl-spo-hyper",
|
10 |
torch_dtype=torch.float16,
|
11 |
)
|
12 |
pipe.to(device)
|
|
|
43 |
}
|
44 |
try:
|
45 |
images = pipe(
|
46 |
+
prompt=prompt + ", masterpiece, best quality, very aesthetic, absurdres",
|
47 |
+
negative_prompt=neg_prompt + ", lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract], photo, deformed, disfigured, low contrast, photo, deformed, disfigured, low contrast",
|
48 |
width=1024,
|
49 |
height=1024,
|
50 |
guidance_scale=7.5,
|