End of training

README.md

@@ -1,19 +1,20 @@
 ---
-license: apache-2.0
-base_model: HuggingFaceM4/idefics2-8b
+license: other
+base_model: HuggingFaceM4/idefics-9b-instruct
 tags:
 - generated_from_trainer
 model-index:
-- name: idefics2-frozenlake
+- name: IDEFICS-frozenlake
   results: []
+library_name: peft
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# idefics2-frozenlake
+# IDEFICS-frozenlake
 
-This model is a fine-tuned version of [HuggingFaceM4/idefics2-8b](https://huggingface.co/HuggingFaceM4/idefics2-8b) on an unknown dataset.
+This model is a fine-tuned version of [HuggingFaceM4/idefics-9b-instruct](https://huggingface.co/HuggingFaceM4/idefics-9b-instruct) on an unknown dataset.
 
 ## Model description
 
@@ -29,6 +30,21 @@ More information needed
 
 ## Training procedure
 
+
+The following `bitsandbytes` quantization config was used during training:
+- quant_method: bitsandbytes
+- _load_in_8bit: False
+- _load_in_4bit: True
+- llm_int8_threshold: 6.0
+- llm_int8_skip_modules: ['lm_head', 'embed_tokens']
+- llm_int8_enable_fp32_cpu_offload: False
+- llm_int8_has_fp16_weight: False
+- bnb_4bit_quant_type: nf4
+- bnb_4bit_use_double_quant: True
+- bnb_4bit_compute_dtype: bfloat16
+- bnb_4bit_quant_storage: uint8
+- load_in_4bit: True
+- load_in_8bit: False
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
@@ -50,7 +66,8 @@ The following hyperparameters were used during training:
 
 ### Framework versions
 
-- Transformers 4.41.2
-- Pytorch 2.1.1+cu121
-- Datasets 2.20.0
+- PEFT 0.5.0
+- Transformers 4.43.3
+- Pytorch 2.4.0+cu121
+- Datasets 2.16.1
 - Tokenizers 0.19.1