Skip to content

Commit

Permalink
Fix pose preprocessor not resulting in a vector layer
Browse files Browse the repository at this point in the history
  • Loading branch information
Acly committed Oct 20, 2024
1 parent b5b0b1a commit 022bc24
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ai_diffusion/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -606,7 +606,7 @@ def apply_generated_result(self, job_id: str, index: int):

def add_control_layer(self, job: Job, result: dict | SharedWorkflow | None):
assert job.kind is JobKind.control_layer and job.control
if job.control.mode is ControlMode.pose and isinstance(result, dict):
if job.control.mode is ControlMode.pose and isinstance(result, (dict, list)):
pose = Pose.from_open_pose_json(result)
pose.scale(job.params.bounds.extent)
return self.layers.create_vector(job.params.name, pose.to_svg())
Expand Down

0 comments on commit 022bc24

Please sign in to comment.