Skip to content

Commit aaa870e

Browse files
committed
code format changes
Signed-off-by: Molly Sophia <mollysophia379@gmail.com>
1 parent f2c1a5c commit aaa870e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

convert_hf_to_gguf.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3258,6 +3258,7 @@ def set_gguf_parameters(self):
32583258
self.gguf_writer.add_head_count(0)
32593259

32603260
lerp_weights: dict[int, dict[str, Tensor]] = {}
3261+
32613262
def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iterable[tuple[str, Tensor]]:
32623263
new_name = self.map_tensor_name(name)
32633264

@@ -3283,7 +3284,7 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
32833284

32843285
# concat time_mix_lerp weights to reduce some cpu overhead
32853286
# also reduces the number of tensors in the model
3286-
if bid is not None and "time_mix_lerp" in new_name and not "time_mix_lerp_x" in new_name:
3287+
if bid is not None and "time_mix_lerp" in new_name and "time_mix_lerp_x" not in new_name:
32873288
try:
32883289
self.lerp_weights[bid][new_name] = data_torch
32893290
except KeyError:

0 commit comments

Comments
 (0)