Black auto-format

octoml · Mar 1, 2024 · 1b5620e · 1b5620e
1 parent 677fe40
commit 1b5620e
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/mlc_llm/core.py b/mlc_llm/core.py
@@ -522,9 +522,7 @@ def _gen_model_name():
     artifact_tag = parsed.artifact_tag if parsed.artifact_tag else "-".join(_gen_model_name())
     parsed.artifact_path = os.path.join(parsed.artifact_path, artifact_tag)
 
-    parsed.lib_name = (
-        f"{os.path.split(parsed.model)[1]}-{parsed.quantization.name}-{parsed.target_kind}.{parsed.lib_format}"
-    )
+    parsed.lib_name = f"{os.path.split(parsed.model)[1]}-{parsed.quantization.name}-{parsed.target_kind}.{parsed.lib_format}"
     parsed.lib_path = os.path.join(parsed.artifact_path, parsed.lib_name)
 
     return parsed
@@ -1200,7 +1198,6 @@ def as_float16(A: R.Tensor(shape)) -> R.Tensor(shape, dtype="float16"):
             as_float16 = bb.add_func(as_float16, f"as_float16_{ndim}d")
             return as_float16(arg)
 
-
         float16_params = manager_transform["transform_params"].params
         either_float16_params = [
             relax.Var(
@@ -1225,7 +1222,6 @@ def as_float16(A: R.Tensor(shape)) -> R.Tensor(shape, dtype="float16"):
 
         parameter_transforms.append(bb.get())
 
-
     # If the model provided constants, bind them to the transform_params.
     # This ensures that the generated function can reproduce the same
     # output parameters even when executed outside of the `build.py`