Spaces:

John6666
/

text2tag-llm

Running on Zero

John6666 commited on Jul 28, 2024

Commit

71c4242

verified ·

1 Parent(s): e2dcb7f

Upload genimage.py

Files changed (1) hide show

genimage.py CHANGED Viewed

@@ -6,7 +6,7 @@ def load_pipeline():
     import torch
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     pipe = StableDiffusionXLPipeline.from_pretrained(
-        "John6666/t-ponynai3-v6-sdxl-spo-hyper",
         torch_dtype=torch.float16,
     )
     pipe.to(device)

     import torch
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     pipe = StableDiffusionXLPipeline.from_pretrained(
+        "John6666/mala-anime-mix-nsfw-pony-xl-v6-sdxl-spo-pcm",
         torch_dtype=torch.float16,
     )
     pipe.to(device)