John6666 commited on
Commit
e2dcb7f
·
verified ·
1 Parent(s): 3f700b5

Upload genimage.py

Browse files
Files changed (1) hide show
  1. 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/comradeship-xl-v9se2-sdxl",
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 + ", score_9, score_8_up, score_7_up, score_6_up, score_5_up, score_4_up, masterpiece, best quality, very aesthetic, absurdres",
47
- negative_prompt=neg_prompt + ", score_4, score_3, score_2, score_1, bad hands, bad feet, 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,
 
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,