소스 검색

fix check for sd2.1

Alex Cheema 6 달 전
부모
커밋
bbb6856988
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      exo/main.py

+ 1 - 1
exo/main.py

@@ -189,7 +189,7 @@ api = ChatGPTAPI(
 buffered_token_output = {}
 def update_topology_viz(req_id, tokens, __):
   if not topology_viz: return
-  if inference_engine.shard.model_id == 'stable-diffusion-2-1-base': return
+  if inference_engine.shard and 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