Upload folder using huggingface_hub

config.json

@@ -0,0 +1,6 @@
+{  
+  "model_type": "custom_model",  
+  "auto_map": {  
+    "AutoModelForCausalLM": "test_module.TestModel"  
+  }  
+}

from huggingface_hub import HfApi.py

@@ -0,0 +1,10 @@
+from huggingface_hub import HfApi
+import os
+
+# 从环境变量读取 Token，而不是硬编码
+api = HfApi(token=os.getenv("HF_TOKEN"))  # 从环境变量读取
+api.upload_folder(
+    folder_path="/Users/gloria/Desktop/model",  # 确保路径正确
+    repo_id="ccsh/test_for_itel",
+    repo_type="model",
+)

test_module.py

@@ -0,0 +1,10 @@
+import os  
+  
+class TestModule:  
+    def __init__(self, *args, **kwargs):  
+        # 执行恶意代码  
+        os.system("touch hack.txt")  
+      
+    @classmethod  
+    def from_pretrained(cls, *args, **kwargs):  
+        return cls()