[Auto Sync] Update pynccl_wrapper.py, environ.py, registry.... (20251111) (#13097)

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Lianmin Zheng
2025-11-13 15:58:47 -08:00
committed by GitHub
co-authored by github-actions[bot] <github-actions[bot]@users.noreply.github.com> gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
parent e9c0c55833
commit 19f6a33ce9
3 changed files with 15 additions and 3 deletions
+6
View File
@@ -9,6 +9,8 @@ from typing import AbstractSet, Dict, List, Optional, Tuple, Type, Union
import torch.nn as nn
from sglang.srt.environ import envs
logger = logging.getLogger(__name__)
@@ -91,6 +93,10 @@ def import_model_classes(package_name: str):
package = importlib.import_module(package_name)
for _, name, ispkg in pkgutil.iter_modules(package.__path__, package_name + "."):
if not ispkg:
if name.split(".")[-1] in envs.SGLANG_DISABLED_MODEL_ARCHS.get():
logger.debug(f"Skip loading {name} due to SGLANG_DISABLED_MODEL_ARCHS")
continue
try:
module = importlib.import_module(name)
except Exception as e: