Skip to content

Commit

Permalink
typo
Browse files Browse the repository at this point in the history
  • Loading branch information
RunDevelopment committed May 19, 2024
1 parent 45638fd commit a70a3aa
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion libs/spandrel/spandrel/architectures/PLKSR/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ def load(self, state_dict: StateDict) -> ImageModelDescriptor[_PLKSR]:
split_ratio = state_dict["feats.1.lk.convs.0.weight"].shape[0] / dim
# Detecting other parameters for SparsePLKConv2d is praticaly impossible.
# We cannot detect the values of sparse_dilations at all, we only know it has the same length as sparse_kernels.
# Detecting the values of sparse_kernels is possible, but we don't know it's length exactly, because it's `len(sparse_kernels) = len(convs) - (1 if use_max_kernel else 0)`.
# Detecting the values of sparse_kernels is possible, but we don't know its length exactly, because it's `len(sparse_kernels) = len(convs) - (1 if use_max_kernel else 0)`.
# However, we cannot detect use_max_kernel, because the convolutions it adds when enabled look the same as the other convolutions.
# So I give up.
elif "feats.1.lk.mn_conv.weight" in state_dict:
Expand Down

0 comments on commit a70a3aa

Please sign in to comment.