format

huggingface · Jul 22, 2024 · 2c561d3 · 2c561d3
1 parent 6ef2081
commit 2c561d3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/optimum/fx/parallelization/passes.py b/optimum/fx/parallelization/passes.py
@@ -514,6 +514,7 @@ def run(self, graph_module: GraphModule, ctx: ParallelExecutionCtx, config: Conf
                 # weights loading
                 if target.source in ctx.weight_map:
                     from safetensors import safe_open
+
                     with safe_open(ctx.weight_map[target.source], framework="pt", device="cpu") as fp:
                         tensor_slice = fp.get_slice(target.source)
                         source_index = [