File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -146,11 +146,10 @@ def register_router():
146146 from vllm_fl .ops .fused_moe .router import replace_router_with_fl
147147 replace_router_with_fl ()
148148
149-
150-
151149def register_model ():
152150 """Register FL-specific models not yet upstream."""
153151 from vllm import ModelRegistry
152+
154153 _register_flagcx_connector ()
155154
156155 # Register OOT quant kernels so kernel selection can find them
Original file line number Diff line number Diff line change 88try :
99 # FlagGems<=5.0.2: DeviceDetector lives in device.
1010 from flag_gems .runtime .backend .device import DeviceDetector
11- except ImportError :
11+ except ( ImportError , FileNotFoundError ) :
1212 # FlagGems>5.0.2: DeviceDetector lives in device_finder.
1313 from flag_gems .runtime .backend .device_finder import DeviceDetector
1414from flag_gems .runtime import backend
You can’t perform that action at this time.
0 commit comments