amiguel commited on
Commit
42e3261
·
verified ·
1 Parent(s): 663f974

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -275,8 +275,7 @@ def create_model(src_vocab, tgt_vocab, N, d_model, d_ff, h, dropout=0.1):
275
  pos = PositionalEncoding(d_model, dropout).to(DEVICE)
276
  model = Transformer(
277
  Encoder(EncoderLayer(d_model, deepcopy(attn), deepcopy(ff), dropout).to(DEVICE), N).to(DEVICE),
278
- Decoder(DecoderLayer(d_model, deepcopy(attn), deepcopy(attn), deepcopy(ff), dropout).to(DEVICE), N).to(DEVIC
279
- E),
280
  nn.Sequential(Embeddings(d_model, src_vocab).to(DEVICE), deepcopy(pos)),
281
  nn.Sequential(Embeddings(d_model, tgt_vocab).to(DEVICE), deepcopy(pos)),
282
  Generator(d_model, tgt_vocab)).to(DEVICE)
 
275
  pos = PositionalEncoding(d_model, dropout).to(DEVICE)
276
  model = Transformer(
277
  Encoder(EncoderLayer(d_model, deepcopy(attn), deepcopy(ff), dropout).to(DEVICE), N).to(DEVICE),
278
+ Decoder(DecoderLayer(d_model, deepcopy(attn), deepcopy(attn), deepcopy(ff), dropout).to(DEVICE), N).to(DEVICE),
 
279
  nn.Sequential(Embeddings(d_model, src_vocab).to(DEVICE), deepcopy(pos)),
280
  nn.Sequential(Embeddings(d_model, tgt_vocab).to(DEVICE), deepcopy(pos)),
281
  Generator(d_model, tgt_vocab)).to(DEVICE)