riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-04 13:20:12 +06:00
Code Issues Actions 21 Packages Projects Releases Wiki Activity

Merge 1a8d584598 into 2d561713f8

Browse Source
This commit is contained in:
kilavvy 2025-07-03 00:01:19 +05:30 committed by GitHub
commit cdb2f0aba0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/swin/modeling_swin.py
View File

@ -341,7 +341,7 @@ class SwinPatchMerging(nn.Module):
batch_size, dim, num_channels = input_feature.shape batch_size, dim, num_channels = input_feature.shape
input_feature = input_feature.view(batch_size, height, width, num_channels) input_feature = input_feature.view(batch_size, height, width, num_channels)
# pad input to be disible by width and height, if needed # pad input to be divisible by width and height, if needed
input_feature = self.maybe_pad(input_feature, height, width) input_feature = self.maybe_pad(input_feature, height, width)
# [batch_size, height/2, width/2, num_channels] # [batch_size, height/2, width/2, num_channels]
input_feature_0 = input_feature[:, 0::2, 0::2, :] input_feature_0 = input_feature[:, 0::2, 0::2, :]

2
src/transformers/pipelines/pt_utils.py
View File

@ -185,7 +185,7 @@ class PipelineChunkIterator(PipelineIterator):
# Try to return next item # Try to return next item
processed = next(self.subiterator) processed = next(self.subiterator)
except StopIteration: except StopIteration:
# When a preprocess iterator ends, we can start lookig at the next item # When a preprocess iterator ends, we can start looking at the next item
# ChunkIterator will keep feeding until ALL elements of iterator # ChunkIterator will keep feeding until ALL elements of iterator
# all have created their subiterator and have been iterating against. # all have created their subiterator and have been iterating against.
# #