Mirrors/bark-with-voice-clone
1
0
Fork 0
mirror of https://github.com/serp-ai/bark-with-voice-clone.git synced 2026-04-03 09:46:24 +02:00
Code Issues Packages Projects Releases Wiki Activity

simplify device placement

Browse Source
This commit is contained in:
Georg Kucsko
2023-04-25 17:49:35 -04:00
parent 97c6019ecd
commit 6c26fb7b34
2 changed files with 82 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
bark/model.py
View File

@@ -200,7 +200,6 @@ class GPT(nn.Module):
pos_emb = self.transformer.wpe(position_ids) # position embeddings of shape (1, t, n_embd)
x = self.transformer.drop(tok_emb + pos_emb)
new_kv = () if use_cache else None