Create download_file.py

download_file.py

@@ -0,0 +1,19 @@
+# download_model.py
+import os
+from unittest.mock import patch
+from transformers import AutoModelForCausalLM, AutoProcessor
+from transformers.dynamic_module_utils import get_imports
+
+def fixed_get_imports(filename: os.PathLike) -> list[str]:
+    if not str(filename).endswith("/modeling_florence2.py"):
+        return get_imports(filename)
+    imports = get_imports(filename)
+    if "flash_attn" in imports:
+        imports.remove("flash_attn")
+    return imports
+
+with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports):
+    model = AutoModelForCausalLM.from_pretrained("microsoft/Florence-2-base-ft", trust_remote_code=True)
+    processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base-ft", trust_remote_code=True)
+    model.save_pretrained("/code/florence_model")
+    processor.save_pretrained("/code/florence_processor")