Spaces:

yeq6x
/

PositionMapVisualizer

Sleeping

yeq6x commited on Jan 1

Commit

2867bc8

1 Parent(s): 7608cfe

dependencies

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,12 +35,12 @@ ratios = np.array(list(ratios_map.keys()))
 openpose = OpenposeDetector.from_pretrained('lllyasviel/ControlNet')
 controlnet = ControlNetModel.from_pretrained(
-    "briaai/BRIA-2.3-ControlNet-Pose",
     torch_dtype=torch.float16
 ).to('cuda')
 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
-    "briaai/BRIA-2.3",
     controlnet=controlnet,
     torch_dtype=torch.float16,
     low_cpu_mem_usage=True,

 openpose = OpenposeDetector.from_pretrained('lllyasviel/ControlNet')
 controlnet = ControlNetModel.from_pretrained(
+    "yeq6x/Image2PositionColor_v3",
     torch_dtype=torch.float16
 ).to('cuda')
 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
+    "yeq6x/animagine_position_map",
     controlnet=controlnet,
     torch_dtype=torch.float16,
     low_cpu_mem_usage=True,