Skip to content

Commit

Permalink
test
Browse files Browse the repository at this point in the history
  • Loading branch information
db0 committed Sep 13, 2024
1 parent c2f6bc4 commit ba20af2
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 1 deletion.
1 change: 1 addition & 0 deletions horde/bridge_reference.py
Original file line number Diff line number Diff line change
Expand Up @@ -186,6 +186,7 @@ def parse_bridge_agent(bridge_agent):
@logger.catch(reraise=True)
def check_bridge_capability(capability, bridge_agent):
bridge_name, bridge_version = parse_bridge_agent(bridge_agent)
logger.debug([bridge_name, bridge_version])
if bridge_name not in BRIDGE_CAPABILITIES:
return False
total_capabilities = set()
Expand Down
3 changes: 2 additions & 1 deletion horde/classes/stable/worker.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,11 +140,12 @@ def can_generate(self, waiting_prompt):
and not check_bridge_capability("stable_cascade_2pass", self.bridge_agent)
):
return [False, "bridge_version"]
logger.debug('test')
if "flux_1" in model_reference.get_all_model_baselines(self.get_model_names()) and not check_bridge_capability(
"flux", self.bridge_agent
):
logger.debug('flux')
return [False, "bridge_version"]
logger.debug('test')
if waiting_prompt.params.get("clip_skip", 1) > 1 and not check_bridge_capability(
"clip_skip",
self.bridge_agent,
Expand Down

0 comments on commit ba20af2

Please sign in to comment.