alihan/exo
1
0
Fork 0
mirror of https://github.com/exo-explore/exo.git synced 2025-10-23 02:57:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix check for sd2.1

Browse Source
This commit is contained in:
Alex Cheema
2025-01-22 23:51:09 +00:00
parent 9ba8bbbcf8
commit bbb6856988
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
exo/main.py
View File

@@ -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