Spaces:

yeq6x
/

PositionMapVisualizer

Sleeping

yeq6x commited on Jan 1

Commit

7351881

1 Parent(s): fab2b8a

load_model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,9 +18,6 @@ def load_model():
     pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
         "yeq6x/animagine_position_map",
         controlnet=ControlNetModel.from_pretrained("yeq6x/Image2PositionColor_v3"),
-        # torch_dtype=torch.float16,
-        use_safetensors=True,
-        # variant="fp16"
     ).to("cuda")
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)

     pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
         "yeq6x/animagine_position_map",
         controlnet=ControlNetModel.from_pretrained("yeq6x/Image2PositionColor_v3"),
     ).to("cuda")
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)