Update vision_transformer_hybrid.py

Smartappli · May 25, 2024 · a906eba · a906eba
1 parent 15401ee
commit a906eba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fl_common/models/vision_transformer_hybrid.py b/fl_common/models/vision_transformer_hybrid.py
@@ -45,7 +45,7 @@ def get_vision_transformer_hybrid_model(
         )
     except RuntimeError as e:
         print(
-            f"{vision_transformer_hybrid_type} - Error loading pretrained model: {e}"
+            f"{vision_transformer_hybrid_type} - Error loading pretrained model: {e}",
         )
         return create_model(
             vision_transformer_hybrid_type,