format

vllm-project · Oct 5, 2024 · 45707e2 · 45707e2
1 parent d9f86a9
commit 45707e2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vllm/_custom_ops.py b/vllm/_custom_ops.py
@@ -3,6 +3,7 @@
 from typing import List, Optional, Tuple, Union
 
 import torch
+import torch.library
 
 import vllm.envs as envs
 from vllm._core_ext import ScalarType
@@ -25,12 +26,12 @@
     import vllm._moe_C  # noqa: F401
     supports_moe_ops = True
 
-import torch.library
 try:
     import torch.library.register_fake
 except ImportError:
     from torch.library import impl_abstract as register_fake
 
+
 def hint_on_error(fn):
 
     @functools.wraps(fn)