fix check for sd2.1

2025-01-22 23:53:42 +00:00
parent bbb6856988
commit f8db4e131e
1 changed files with 2 additions and 1 deletions
@@ -189,7 +189,8 @@ api = ChatGPTAPI(
 buffered_token_output = {}
 def update_topology_viz(req_id, tokens, __):
  if not topology_viz: return
-  if inference_engine.shard and inference_engine.shard.model_id == 'stable-diffusion-2-1-base': return
+  if not inference_engine.shard: return
+  if inference_engine.shard.model_id == 'stable-diffusion-2-1-base': return

  if req_id in buffered_token_output: buffered_token_output[req_id].extend(tokens)
  else: buffered_token_output[req_id] = tokens