Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
Signed-off-by: Vladimir Mandic <[email protected]>
  • Loading branch information
vladmandic committed Dec 21, 2024
1 parent 29c76cd commit 8924e5b
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 4 deletions.
4 changes: 1 addition & 3 deletions modules/processing_class.py
Original file line number Diff line number Diff line change
Expand Up @@ -348,15 +348,13 @@ def __init__(self, **kwargs):
debug(f'Process init: mode={self.__class__.__name__} kwargs={kwargs}') # pylint: disable=protected-access
super().__init__(**kwargs)

def init(self, all_prompts=None, all_negative_prompts=None, all_seeds=None, all_subseeds=None):
def init(self, all_prompts=None, all_seeds=None, all_subseeds=None):
if shared.native:
shared.sd_model = sd_models.set_diffuser_pipe(self.sd_model, sd_models.DiffusersTaskType.TEXT_2_IMAGE)
self.width = self.width or 1024
self.height = self.height or 1024
if all_prompts is not None:
self.all_prompts = all_prompts
if all_negative_prompts is not None:
self.all_negative_prompts = all_negative_prompts
if all_seeds is not None:
self.all_seeds = all_seeds
if all_subseeds is not None:
Expand Down
2 changes: 1 addition & 1 deletion modules/prompt_parser.py
Original file line number Diff line number Diff line change
Expand Up @@ -147,7 +147,7 @@ def __default__(self, data, children, meta):
def get_schedule(prompt):
try:
tree = schedule_parser.parse(prompt)
except lark.exceptions.LarkError:
except Exception:
return [[steps, prompt]]
return [[t, at_step(t, tree)] for t in collect_steps(steps, tree)]

Expand Down

0 comments on commit 8924e5b

Please sign in to comment.