From ecc4f1bdfae5b3a9679fae499c5c9b375d927547 Mon Sep 17 00:00:00 2001
From: LysandreJik <lysandre.debut@reseau.eseo.fr>
Date: Thu, 3 Oct 2019 17:42:16 -0400
Subject: [PATCH] XLM use_lang_embedding flag in run_generation

---
 examples/run_generation.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples/run_generation.py b/examples/run_generation.py
index a70a0e7842..83926f42b7 100644
--- a/examples/run_generation.py
+++ b/examples/run_generation.py
@@ -177,7 +177,8 @@ def main():
     while True:
         xlm_lang = None
         # XLM Language usage detailed in the issues #1414
-        if args.model_type in ["xlm"] and hasattr(tokenizer, 'lang2id'):
+        if args.model_type in ["xlm"] and hasattr(tokenizer, 'lang2id') and hasattr(model.config, 'use_lang_emb') \
+                and model.config.use_lang_emb:
             if args.xlm_lang:
                 language = args.xlm_lang
             else: