"...git@developer.sourcefind.cn:modelzoo/solov2-pytorch.git" did not exist on "b4af1c941ac805aa7b8033805d2e923a67fbea21"
Unverified Commit 4f7b434a authored by Sergei Belousov's avatar Sergei Belousov Committed by GitHub
Browse files

Update modeling_bark.py (#30221)

Change .view() to .reshape() to prevent errors on non-contiguous tensors
parent bf9a7ab9
...@@ -1068,7 +1068,7 @@ class BarkCoarseModel(BarkCausalModel): ...@@ -1068,7 +1068,7 @@ class BarkCoarseModel(BarkCausalModel):
x_coarse_history[n, :] += codebook_size * n x_coarse_history[n, :] += codebook_size * n
# flatten x_coarse_history # 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 x_coarse_history = x_coarse_history + semantic_generation_config.semantic_vocab_size
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment