phi-2-mental-health

README.md

@@ -6,6 +6,7 @@ tags:
 model-index:
 - name: results
   results: []
+library_name: peft
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -29,6 +30,28 @@ More information needed
 
 ## Training procedure
 
+
+The following `bitsandbytes` quantization config was used during training:
+- load_in_8bit: False
+- load_in_4bit: True
+- llm_int8_threshold: 6.0
+- llm_int8_skip_modules: None
+- llm_int8_enable_fp32_cpu_offload: False
+- llm_int8_has_fp16_weight: False
+- bnb_4bit_quant_type: nf4
+- bnb_4bit_use_double_quant: False
+- bnb_4bit_compute_dtype: float16
+
+The following `bitsandbytes` quantization config was used during training:
+- load_in_8bit: False
+- load_in_4bit: True
+- llm_int8_threshold: 6.0
+- llm_int8_skip_modules: None
+- llm_int8_enable_fp32_cpu_offload: False
+- llm_int8_has_fp16_weight: False
+- bnb_4bit_quant_type: nf4
+- bnb_4bit_use_double_quant: False
+- bnb_4bit_compute_dtype: float16
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
@@ -49,6 +72,8 @@ The following hyperparameters were used during training:
 
 ### Framework versions
 
+- PEFT 0.4.0
+- PEFT 0.4.0
 - Transformers 4.31.0
 - Pytorch 2.1.0+cu118
 - Datasets 2.14.6

adapter_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7b55dacc7d46b7bcdcafcdd51b0eb0de903c10a017700d3f27132f06086ba0cf
+oid sha256:99db4ec6db414ba78802cc285add23f608b9b92d7b8937a04bc919ed5799dfb4
 size 36768266

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b7aabef4ed84be39b641b1a53b9f5fd55fad20237eef75ed5a864f5c1b5d5744
+oid sha256:844e38396fe1f532594cae81c4be7430436893214b3f914cba5cb8a459249d0b
 size 4344