Upload folder using huggingface_hub
Browse files- README.md +89 -3
- adapter_config.json +31 -0
- adapter_model.safetensors +3 -0
- config.json +33 -0
- merges.txt +0 -0
- special_tokens_map.json +24 -0
- tokenizer.json +0 -0
- tokenizer_config.json +21 -0
- vocab.json +0 -0
README.md
CHANGED
@@ -1,3 +1,89 @@
|
|
1 |
-
---
|
2 |
-
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: tiiuae/falcon-rw-1b
|
3 |
+
library_name: peft
|
4 |
+
license: apache-2.0
|
5 |
+
datasets:
|
6 |
+
- marmikpandya/mental-health
|
7 |
+
language:
|
8 |
+
- en
|
9 |
+
tags:
|
10 |
+
- mentalhealth
|
11 |
+
- selfcare
|
12 |
+
- wellness
|
13 |
+
- wellbeing
|
14 |
+
- depression
|
15 |
+
- anxiety
|
16 |
+
- stress
|
17 |
+
- emotionalsupport
|
18 |
+
- mentalsupport
|
19 |
+
- advisor
|
20 |
+
---
|
21 |
+
|
22 |
+
# Model Card for Model ID
|
23 |
+
|
24 |
+
Falcon-1B-Mental-Health-Advisor is a fine-tuned version of the tiiuae/falcon-rw-1b model, adapted for providing empathetic and contextually relevant responses to mental health-related queries. The model has been trained on a curated dataset to assist in mental health conversations, offering advice, guidance, and support for individuals dealing with issues like stress, anxiety, and depression. It provides a compassionate approach to mental health queries while focusing on promoting emotional well-being and mental health awareness.
|
25 |
+
|
26 |
+
## Model Details
|
27 |
+
|
28 |
+
# Falcon-1B Fine-Tuned for Mental Health (LoRA)
|
29 |
+
|
30 |
+
This is a LoRA adapter for the Falcon-RW-1B model. It was fine-tuned on the 'marmikpandya/mental-health' dataset.
|
31 |
+
|
32 |
+
## Usage
|
33 |
+
|
34 |
+
Since this model is an adapter, it **must** be loaded with the original Falcon-RW-1B model using PEFT:
|
35 |
+
|
36 |
+
```python
|
37 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
38 |
+
from peft import PeftModel
|
39 |
+
|
40 |
+
base_model = "tiiuae/falcon-rw-1b"
|
41 |
+
model = AutoModelForCausalLM.from_pretrained(base_model, device_map="auto")
|
42 |
+
model = PeftModel.from_pretrained(model, "ShivomH/Falcon-1B-Mental-Health-Finetuned")
|
43 |
+
|
44 |
+
tokenizer = AutoTokenizer.from_pretrained("ShivomH/Falcon-1B-Mental-Health-Finetuned")
|
45 |
+
|
46 |
+
### Model Description
|
47 |
+
|
48 |
+
<!-- Provide a longer summary of what this model is. -->
|
49 |
+
|
50 |
+
- **Developed by:** Shivom Hatalkar
|
51 |
+
- **Model type:** Text-Generation
|
52 |
+
- **Language(s) (NLP):** English
|
53 |
+
- **License:** apache-2.0
|
54 |
+
- **Finetuned on model:** falcon-rw-1b
|
55 |
+
|
56 |
+
### Model Sources [optional]
|
57 |
+
|
58 |
+
<!-- Provide the basic links for the model. -->
|
59 |
+
|
60 |
+
- **Repository:** [More Information Needed]
|
61 |
+
- **Paper [optional]:** [More Information Needed]
|
62 |
+
- **Demo [optional]:** [More Information Needed]
|
63 |
+
|
64 |
+
## Bias, Risks, and Limitations
|
65 |
+
|
66 |
+
Not a Substitute for Professional Care - This model is not a licensed mental health professional. Its responses may be incomplete, inaccurate, or unsuitable for serious conditions.
|
67 |
+
|
68 |
+
Inherent Biases - May reflect biases in training data (e.g., cultural assumptions, stigmatizing language).
|
69 |
+
|
70 |
+
Crisis Limitations - Not designed for crisis intervention (e.g., suicidal ideation, self-harm). Always direct users to human professionals or emergency services.
|
71 |
+
|
72 |
+
Over-Reliance Risk - Outputs could inadvertently worsen symptoms if users interpret them as definitive advice.
|
73 |
+
|
74 |
+
Privacy - Avoid sharing sensitive personal details, as generated text is not anonymized or secure.
|
75 |
+
|
76 |
+
Intended Use - Assist with general emotional support, not diagnosis or treatment.
|
77 |
+
|
78 |
+
## How to Get Started with the Model
|
79 |
+
|
80 |
+
Use the code below to get started with the model.
|
81 |
+
|
82 |
+
[More Information Needed]
|
83 |
+
|
84 |
+
## Training Details
|
85 |
+
|
86 |
+
### Framework versions
|
87 |
+
|
88 |
+
- PEFT 0.14.0
|
89 |
+
```
|
adapter_config.json
ADDED
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "tiiuae/falcon-rw-1b",
|
5 |
+
"bias": "none",
|
6 |
+
"eva_config": null,
|
7 |
+
"exclude_modules": null,
|
8 |
+
"fan_in_fan_out": false,
|
9 |
+
"inference_mode": true,
|
10 |
+
"init_lora_weights": true,
|
11 |
+
"layer_replication": null,
|
12 |
+
"layers_pattern": null,
|
13 |
+
"layers_to_transform": null,
|
14 |
+
"loftq_config": {},
|
15 |
+
"lora_alpha": 32,
|
16 |
+
"lora_bias": false,
|
17 |
+
"lora_dropout": 0.2,
|
18 |
+
"megatron_config": null,
|
19 |
+
"megatron_core": "megatron.core",
|
20 |
+
"modules_to_save": null,
|
21 |
+
"peft_type": "LORA",
|
22 |
+
"r": 32,
|
23 |
+
"rank_pattern": {},
|
24 |
+
"revision": null,
|
25 |
+
"target_modules": [
|
26 |
+
"query_key_value"
|
27 |
+
],
|
28 |
+
"task_type": "CAUSAL_LM",
|
29 |
+
"use_dora": false,
|
30 |
+
"use_rslora": false
|
31 |
+
}
|
adapter_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0c82195311d4cccae519892ea540669890fa25d3d1ef698b2431391ca827e62f
|
3 |
+
size 25173000
|
config.json
ADDED
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alibi": true,
|
3 |
+
"apply_residual_connection_post_layernorm": false,
|
4 |
+
"architectures": [
|
5 |
+
"FalconForCausalLM"
|
6 |
+
],
|
7 |
+
"attention_dropout": 0.0,
|
8 |
+
"auto_map": {
|
9 |
+
"AutoConfig": "configuration_falcon.FalconConfig",
|
10 |
+
"AutoModel": "modeling_falcon.FalconModel",
|
11 |
+
"AutoModelForSequenceClassification": "modeling_falcon.FalconForSequenceClassification",
|
12 |
+
"AutoModelForTokenClassification": "modeling_falcon.FalconForTokenClassification",
|
13 |
+
"AutoModelForQuestionAnswering": "modeling_falcon.FalconForQuestionAnswering",
|
14 |
+
"AutoModelForCausalLM": "modeling_falcon.FalconForCausalLM"
|
15 |
+
},
|
16 |
+
"bias": true,
|
17 |
+
"bos_token_id": 1,
|
18 |
+
"eos_token_id": 2,
|
19 |
+
"hidden_dropout": 0.0,
|
20 |
+
"hidden_size": 2048,
|
21 |
+
"initializer_range": 0.02,
|
22 |
+
"layer_norm_epsilon": 1e-05,
|
23 |
+
"model_type": "falcon",
|
24 |
+
"multi_query": false,
|
25 |
+
"new_decoder_architecture": false,
|
26 |
+
"num_attention_heads": 32,
|
27 |
+
"num_hidden_layers": 24,
|
28 |
+
"parallel_attn": false,
|
29 |
+
"torch_dtype": "bfloat16",
|
30 |
+
"transformers_version": "4.27.4",
|
31 |
+
"use_cache": true,
|
32 |
+
"vocab_size": 50304
|
33 |
+
}
|
merges.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|
special_tokens_map.json
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"bos_token": {
|
3 |
+
"content": "<|endoftext|>",
|
4 |
+
"lstrip": false,
|
5 |
+
"normalized": false,
|
6 |
+
"rstrip": false,
|
7 |
+
"single_word": false
|
8 |
+
},
|
9 |
+
"eos_token": {
|
10 |
+
"content": "<|endoftext|>",
|
11 |
+
"lstrip": false,
|
12 |
+
"normalized": false,
|
13 |
+
"rstrip": false,
|
14 |
+
"single_word": false
|
15 |
+
},
|
16 |
+
"pad_token": "<|endoftext|>",
|
17 |
+
"unk_token": {
|
18 |
+
"content": "<|endoftext|>",
|
19 |
+
"lstrip": false,
|
20 |
+
"normalized": false,
|
21 |
+
"rstrip": false,
|
22 |
+
"single_word": false
|
23 |
+
}
|
24 |
+
}
|
tokenizer.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
tokenizer_config.json
ADDED
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_prefix_space": false,
|
3 |
+
"added_tokens_decoder": {
|
4 |
+
"50256": {
|
5 |
+
"content": "<|endoftext|>",
|
6 |
+
"lstrip": false,
|
7 |
+
"normalized": false,
|
8 |
+
"rstrip": false,
|
9 |
+
"single_word": false,
|
10 |
+
"special": true
|
11 |
+
}
|
12 |
+
},
|
13 |
+
"bos_token": "<|endoftext|>",
|
14 |
+
"clean_up_tokenization_spaces": true,
|
15 |
+
"eos_token": "<|endoftext|>",
|
16 |
+
"extra_special_tokens": {},
|
17 |
+
"model_max_length": 1024,
|
18 |
+
"pad_token": "<|endoftext|>",
|
19 |
+
"tokenizer_class": "GPT2Tokenizer",
|
20 |
+
"unk_token": "<|endoftext|>"
|
21 |
+
}
|
vocab.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|