Skip to content

Commit 511a39a

Browse files
committed
Remove unnecessary modified code
1 parent c5b373f commit 511a39a

2 files changed

Lines changed: 2 additions & 3 deletions

File tree

vllm_fl/__init__.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff 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-
151149
def 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

vllm_fl/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
try:
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
1414
from flag_gems.runtime import backend

0 commit comments

Comments
 (0)