Update modeling_bark.py (#30221)

Change .view() to .reshape() to prevent errors on non-contiguous tensors

Update modeling_bark.py (#30221)
Change .view() to .reshape() to prevent errors on non-contiguous tensors
4f7b434a · Sergei Belousov · GitHub · bf9a7ab9 · 4f7b434a
Unverified Commit 4f7b434a authored Apr 12, 2024 by Sergei Belousov Committed by GitHub Apr 12, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/bark/modeling_bark.py src/transformers/models/bark/modeling_bark.py +1 -1

No files found.
--- a/src/transformers/models/bark/modeling_bark.py
+++ b/src/transformers/models/bark/modeling_bark.py
@@ -1068,7 +1068,7 @@ class BarkCoarseModel(BarkCausalModel):
                    x_coarse_history[n, :] += codebook_size * n

            # flatten x_coarse_history
-            x_coarse_history = torch.transpose(x_coarse_history, 0, 1).view(-1)
+            x_coarse_history = torch.transpose(x_coarse_history, 0, 1).reshape(-1)

            x_coarse_history = x_coarse_history + semantic_generation_config.semantic_vocab_size