app.py
CHANGED
|
@@ -207,7 +207,7 @@ def handle_invoke(text, selected_types):
|
|
| 207 |
results = []
|
| 208 |
futures_to_model = {} # 鍚刦uture銇ㄣ儮銉囥儷銈掔磹銇ャ亼銈嬨仧銈併伄杈炴浉
|
| 209 |
|
| 210 |
-
with ThreadPoolExecutor(max_workers=len(selected_types)) as executor:
|
| 211 |
futures = []
|
| 212 |
|
| 213 |
# 銉嗐偔銈广儓鐢熸垚銉€儑銉伄瀹熻
|
|
|
|
| 207 |
results = []
|
| 208 |
futures_to_model = {} # 鍚刦uture銇ㄣ儮銉囥儷銈掔磹銇ャ亼銈嬨仧銈併伄杈炴浉
|
| 209 |
|
| 210 |
+
with ThreadPoolExecutor(max_workers=len([x for x in TEXT_GENERATION_MODELS + CLASSIFICATION_MODELS if x["type"] in selected_types])) as executor:
|
| 211 |
futures = []
|
| 212 |
|
| 213 |
# 銉嗐偔銈广儓鐢熸垚銉€儑銉伄瀹熻
|