Add SetFit model

README.md

@@ -114,7 +114,7 @@ preds = model("The caption is detailed and")
 
 ### Training Hyperparameters
 - batch_size: (16, 16)
-- num_epochs: (4, 4)
+- num_epochs: (8, 8)
 - max_steps: -1
 - sampling_strategy: oversampling
 - body_learning_rate: (2e-05, 1e-05)
@@ -134,10 +134,15 @@ preds = model("The caption is detailed and")
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
 | 0.1111 | 1    | 0.2791        | -               |
-| 1.0    | 9    | -             | 0.2136          |
-| 2.0    | 18   | -             | 0.2055          |
-| 3.0    | 27   | -             | 0.1965          |
-| 4.0    | 36   | -             | 0.1903          |
+| 1.0    | 9    | -             | 0.2195          |
+| 2.0    | 18   | -             | 0.2068          |
+| 3.0    | 27   | -             | 0.1879          |
+| 4.0    | 36   | -             | 0.1541          |
+| 5.0    | 45   | -             | 0.1141          |
+| 5.5556 | 50   | 0.1874        | -               |
+| 6.0    | 54   | -             | 0.0762          |
+| 7.0    | 63   | -             | 0.0549          |
+| 8.0    | 72   | -             | 0.0482          |
 
 ### Framework Versions
 - Python: 3.11.11

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5745f07bd3a8217762862016906cf2c9aaec5c2839babf66664f682c8bb9e84f
+oid sha256:6b7c0c7337988ae56e2f53f313b3ba41ba5e63d93c4d7a9a7b3114ed19e81459
 size 90864192

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:737e6c8bc539d02329266772cbb4664ed615e4d72338bb045da5d6c55e9a640b
+oid sha256:79d63f03297b92f21180f6916831c598701ce8de4e7c1db85f671958109b8a9c
 size 3935