feat: add code for finetuning moondream

2024-12-08 17:33:37 +00:00
parent 4bf2c89bf8
commit 6b53cb0411
11 changed files with 261 additions and 0 deletions
--- a/resnet/label.py
+++ b/resnet/label.py
@@ -0,0 +1,8 @@
+def label_fake(example):
+    example["is_synthetic"] = 1.0
+    return example
+
+
+def label_real(example):
+    example["is_synthetic"] = 0.0
+    return example