Update safetensors_converter.py

safetensors_converter.py

@@ -14,7 +14,7 @@ def convert(path: Path, half: bool = False, no_ema: bool = False):
     for k, v in state_dict.items():
         if not isinstance(v, torch.Tensor):
             to_remove.append(k)
-        if no_ema and "ema" in k:
+        elif no_ema and "ema" in k:
             to_remove.append(k)
 
     for k in to_remove: