Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
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)
|