Add merged model + processor

requirements.txt

@@ -1,2 +1,5 @@
 torchvision>=0.19
 Pillow>=10
+datasets>=2.19
+peft>=0.10
+safetensors>=0.4

train_model.py

@@ -337,7 +337,13 @@ if __name__ == "__main__":
             # bundle handler and code
             shutil.copy("train_model.py", tmp)
             shutil.copy("handler.py", tmp)
-            Path(tmp, "requirements.txt").write_text("torchvision>=0.19\nPillow>=10\n")
+            Path(tmp, "requirements.txt").write_text("\n".join([
+                "torchvision>=0.19",
+                "Pillow>=10",
+                "datasets>=2.19",
+                "peft>=0.10",
+                "safetensors>=0.4",
+            ]))
 
             HfApi().upload_folder(
                 repo_id=args.huggingface_model_name,