Skip to content

Commit 2ee7493

Browse files
committed
Apply code formatting to pipeline.py
Run make style to format code according to project standards Signed-off-by: ronantakizawa <[email protected]>
1 parent d9a50f8 commit 2ee7493

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

src/llmcompressor/pipelines/sequential/pipeline.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -97,14 +97,9 @@ def _materialize_meta_tensors(obj):
9797
if isinstance(obj, torch.Tensor) and obj.is_meta:
9898
return torch.zeros_like(obj, device=model_device)
9999
elif isinstance(obj, dict):
100-
return {
101-
k: _materialize_meta_tensors(v)
102-
for k, v in obj.items()
103-
}
100+
return {k: _materialize_meta_tensors(v) for k, v in obj.items()}
104101
elif isinstance(obj, (list, tuple)):
105-
return type(obj)(
106-
[_materialize_meta_tensors(x) for x in obj]
107-
)
102+
return type(obj)([_materialize_meta_tensors(x) for x in obj])
108103
return obj
109104

110105
for subgraph_index, subgraph in enumerate(subgraphs):

0 commit comments

Comments
 (0)