Spaces:

John6666
/

text2tag-llm

Running on Zero

John6666 commited on Jul 28, 2024

Commit

48ccfbd

verified ·

1 Parent(s): 45c0511

Upload genimage.py

Files changed (1) hide show

genimage.py CHANGED Viewed

@@ -7,7 +7,7 @@ def load_pipeline():
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     pipe = AutoPipelineForText2Image.from_pretrained(
         "John6666/rae-diffusion-xl-v2-sdxl-spo-pcm",
-        custom_pipeline="lpw_stable_diffusion_xl",
         torch_dtype=torch.float16,
     )
     pipe.to(device)

     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     pipe = AutoPipelineForText2Image.from_pretrained(
         "John6666/rae-diffusion-xl-v2-sdxl-spo-pcm",
+        custom_pipeline="lpw_stable_diffusion",
         torch_dtype=torch.float16,
     )
     pipe.to(device)