Add guard

Lightning-Universe · Oct 20, 2021 · 328e225 · 328e225
1 parent d5992b1
commit 328e225
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lightning_transformers/core/nlp/model.py b/lightning_transformers/core/nlp/model.py
@@ -110,5 +110,6 @@ def load_from_checkpoint(
     ):
         model: HFTransformer = super().load_from_checkpoint(checkpoint_path, map_location, hparams_file, strict)
         # update model with hf_pipeline_kwargs override
-        model._hf_pipeline_kwargs.update(hf_pipeline_kwargs)
+        if hf_pipeline_kwargs is not None:
+            model._hf_pipeline_kwargs.update(hf_pipeline_kwargs)
         return model