fix check for sd2.1

bbb6856988670f75f60c29055f48e8a127d30b6d · 2025-01-22 23:51:09 +0000 · Alex Cheema

Files touched

M exo/main.py

Diff

commit bbb6856988670f75f60c29055f48e8a127d30b6d
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Wed Jan 22 23:51:09 2025 +0000

    fix check for sd2.1
---
 exo/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/exo/main.py b/exo/main.py
index c5e8c529..9d74721c 100644
--- a/exo/main.py
+++ b/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

← 9ba8bbbc fix filter to include 169.254.* since thats what mac uses fo · back to Exo · fix check for sd2.1 f8db4e13 →