Mel-Iza0 commited on
Commit
ac9c4aa
1 Parent(s): 6d5bcb1

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +108 -0
README.md ADDED
@@ -0,0 +1,108 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - pt
4
+ - en
5
+ - es
6
+ license: mit
7
+ library_name: peft
8
+ tags:
9
+ - SFT
10
+ - pipeline
11
+ base_model: mistralai/Mistral-7B-Instruct-v0.2
12
+ model-index:
13
+ - name: Weni/test-folder-upload
14
+ results: []
15
+ ---
16
+
17
+ # Weni/test-folder-upload
18
+
19
+ This model is a fine-tuned version of [mistralai/Mistral-7B-Instruct-v0.2] on the dataset Weni/zeroshot-3.2.0 with the SFT trainer. It is part of the pipeline project for [Weni](https://weni.ai/).
20
+
21
+ It achieves the following results on the evaluation set:
22
+ {'eval_loss': 0.6534676551818848, 'eval_runtime': 206.9318, 'eval_samples_per_second': 13.855, 'eval_steps_per_second': 3.465, 'epoch': 0.02}
23
+
24
+ ## Intended uses & limitations
25
+
26
+ This model has not been trained to avoid specific intructions.
27
+
28
+ ## Training procedure
29
+
30
+ Finetuning was done on the model mistralai/Mistral-7B-Instruct-v0.2 with the following prompt:
31
+
32
+ ```
33
+ Portuguese:
34
+ <s>[INST]Você é muito especialista em classificar a frase do usuário em um chatbot sobre: {context}
35
+ Pare, pense bem e responda com APENAS UM ÚNICO \`id\` da classe que melhor represente a intenção para a frase do usuário de acordo com a análise de seu contexto, responda APENAS com o \`id\` da classe só se você tiver muita certeza e não explique o motivo. Na ausência, falta de informações ou caso a frase do usuário não se enquadre em nenhuma classe, classifique como "-1".
36
+
37
+ # Essas são as Classes com seus Id e Contexto:
38
+ \`\`\`python
39
+ {all_classes}
40
+ \`\`\`
41
+
42
+ # Frase do usuário: {input}
43
+ # Id da Classe: [/INST] \n {output_id} </s>
44
+
45
+
46
+ Spanish:
47
+ <s>[INST]Eres muy experto en clasificar la frase del usuario en un chatbot sobre: {context}
48
+ Deténgase, piense bien y responda con SOLO UN ÚNICO \`id\` de la clase que mejor represente la intención para la frase del usuario de acuerdo con el análisis de su contexto, responda SOLO con el \`id\` de la clase si está muy seguro y no explique el motivo. En ausencia, falta de información o en caso de que la frase del usuario no se ajuste a ninguna clase, clasifique como "-1".
49
+
50
+ # Estas son las Clases con sus Id y Contexto:
51
+ {all_classes}
52
+
53
+ # Frase del usuario: {input}
54
+ # Id de la Clase: [/INST] \n {output_id} </s>
55
+
56
+
57
+ English:
58
+ <s>[INST]You are very expert in classifying the user sentence in a chatbot about: {context}
59
+ Stop, think carefully, and respond with ONLY ONE SINGLE \`id\` of the class that best represents the intention for the user's sentence according to the analysis of its context, respond ONLY with the \`id\` of the class if you are very sure and do not explain the reason. In the absence, lack of information, or if the user's sentence does not fit into any class, classify as "-1".
60
+
61
+ # These are the Classes and its Context:
62
+ {all_classes}
63
+
64
+ # User's sentence: {input}
65
+ # Class Id: [/INST] \n {output_id} </s>
66
+ ```
67
+
68
+ ### Training hyperparameters
69
+
70
+ The following hyperparameters were used during training:
71
+ - learning_rate: 0.0002
72
+ - per_device_train_batch_size: 8
73
+ - per_device_eval_batch_size: 2
74
+ - gradient_accumulation_steps: 1
75
+ - num_gpus: 2
76
+ - total_train_batch_size: 16
77
+ - optimizer: AdamW
78
+ - lr_scheduler_type: cosine
79
+ - num_steps: 30
80
+ - quantization_type: bitsandbytes
81
+ - LoRA: ("\n - bits: 4\n - use_exllama: True\n - device_map: auto\n - use_cache: False\n - lora_r: 8\n - lora_alpha: 16\n - lora_dropout: 0.1\n - bias: none\n - target_modules: ['q_proj', 'k_proj', 'v_proj', 'o_proj']\n - task_type: CAUSAL_LM",)
82
+
83
+ ### Training results
84
+
85
+ ### Framework versions
86
+
87
+ - PEFT 0.8.2
88
+ - git+https://github.com/huggingface/transformers@main
89
+ - datasets==2.17.1
90
+ - peft==0.8.2
91
+ - safetensors==0.4.2
92
+ - evaluate==0.4.1
93
+ - bitsandbytes==0.42
94
+ - huggingface_hub==0.20.3
95
+ - seqeval==1.2.2
96
+ - optimum==1.17.1
97
+ - auto-gptq==0.7.0
98
+ - gpustat==1.1.1
99
+ - deepspeed==0.13.2
100
+ - wandb==0.16.3
101
+ - git+https://github.com/huggingface/trl.git@main
102
+ - git+https://github.com/huggingface/accelerate.git@main
103
+ - coloredlogs==15.0.1
104
+ - traitlets==5.14.1
105
+ - autoawq@https://github.com/casper-hansen/AutoAWQ/releases/download/v0.2.0/autoawq-0.2.0+cu118-cp310-cp310-linux_x86_64.whl
106
+
107
+ ### Hardware
108
+ - Cloud provided: runpod.io