fix check for sd2.1

exo-explore · Jan 22, 2025 · bbb6856 · bbb6856
1 parent 9ba8bbb
commit bbb6856
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/exo/main.py b/exo/main.py
@@ -189,7 +189,7 @@ def configure_uvloop():
 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