Spaces:

Fucius
/

OMG-InstantID

Running on Zero

Fucius commited on Mar 22

Commit

5995597

•

1 Parent(s): f389568

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -256,7 +256,7 @@ def predict_mask(segmentmodel, sam, image, TEXT_PROMPT, segmentType, confidence
     return masks
 def build_model_sd(pretrained_model, controlnet_path, face_adapter, device, prompts, antelopev2_path, width, height, style_lora):
-    controlnet = ControlNetModel.from_pretrained(controlnet_path, torch_dtype=torch.float16)
     pipe = InstantidMultiConceptPipeline.from_pretrained(
         pretrained_model, controlnet=controlnet, torch_dtype=torch.float16).to(device)

     return masks
 def build_model_sd(pretrained_model, controlnet_path, face_adapter, device, prompts, antelopev2_path, width, height, style_lora):
+    controlnet = ControlNetModel.from_pretrained(controlnet_path, torch_dtype=torch.float16).to(device)
     pipe = InstantidMultiConceptPipeline.from_pretrained(
         pretrained_model, controlnet=controlnet, torch_dtype=torch.float16).to(device)