fix style

huggingface · Jul 27, 2023 · 9633fae · 9633fae
1 parent d3ce904
commit 9633fae
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py
@@ -1240,7 +1240,9 @@ def determine_framework(
             framework = "pt"
         elif any(is_tf_weight_file):
             framework = "tf"
-        elif "model_index.json" in all_files and any(file.endswith((pt_weight_extension, safe_weight_extension)) for file in all_files):
+        elif "model_index.json" in all_files and any(
+            file.endswith((pt_weight_extension, safe_weight_extension)) for file in all_files
+        ):
             # stable diffusion case
             framework = "pt"
         else: