Skip to content

Commit

Permalink
Small bug fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
fofr committed Jun 18, 2024
1 parent 383660c commit a3d1be2
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion predict.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,6 +95,7 @@ def predict(
model: str = Input(
choices=SD3_MODELS,
default="sd3_medium_incl_clips_t5xxlfp16.safetensors",
description="Pick whether to use T5-XXL in fp16, fp8 or not at all",
),
width: int = Input(
description="The width of the image",
Expand Down Expand Up @@ -161,7 +162,7 @@ def predict(
),
negative_conditioning_end: float = Input(
description="When the negative conditioning should stop being applied. By default it is disabled.",
le=20,
le=1,
ge=0,
default=0,
),
Expand Down Expand Up @@ -208,6 +209,7 @@ def predict(
)

self.comfyUI.connect()
self.comfyUI.handle_weights(workflow)
self.comfyUI.run_workflow(workflow)

return optimise_images.optimise_image_files(
Expand Down

0 comments on commit a3d1be2

Please sign in to comment.