[Fix] Fix bug caused by merging draggan (#2029)

fix
open-mmlab · Sep 11, 2023 · d140802 · d140802
1 parent 6fda2cc
commit d140802
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmagic/apis/inferencers/base_mmagic_inferencer.py b/mmagic/apis/inferencers/base_mmagic_inferencer.py
@@ -130,7 +130,7 @@ def __call__(self, **kwargs) -> Union[Dict, List[Dict]]:
         Returns:
             Union[Dict, List[Dict]]: Results of inference pipeline.
         """
-        if ('extra_parameters' in kwargs.keys()
+        if ('extra_parameters' in kwargs.keys() and kwargs['extra_parameters']
                 and 'infer_with_grad' in kwargs['extra_parameters'].keys()
                 and kwargs['extra_parameters']['infer_with_grad']):
             results = self.base_call(**kwargs)