more undos

This commit is contained in:
James Betker 2022-04-08 16:31:08 -06:00
parent 032983e2ed
commit 944b4c3335

View File

@ -59,8 +59,8 @@ def build_hf_gpt_transformer(layers, model_dim, heads, max_mel_seq_len, max_text
""" """
from transformers import GPT2Config, GPT2Model from transformers import GPT2Config, GPT2Model
gpt_config = GPT2Config(vocab_size=256, # Unused. gpt_config = GPT2Config(vocab_size=256, # Unused.
n_positions=1, n_positions=max_mel_seq_len+max_text_seq_len,
n_ctx=1, n_ctx=max_mel_seq_len+max_text_seq_len,
n_embd=model_dim, n_embd=model_dim,
n_layer=layers, n_layer=layers,
n_head=heads, n_head=heads,