Update README.md
Browse files
README.md
CHANGED
@@ -11,7 +11,7 @@ tags:
|
|
11 |
- medical
|
12 |
---
|
13 |
|
14 |
-
# Model Card for Model
|
15 |
|
16 |
## Model Details
|
17 |
|
@@ -70,7 +70,7 @@ base_model = AutoModelForCausalLM.from_pretrained(
|
|
70 |
device_map={"": 0}, token=""
|
71 |
)
|
72 |
|
73 |
-
model = PeftModel.from_pretrained(base_model,"Rustamshry/
|
74 |
|
75 |
instruction = """
|
76 |
You are a helpful mental health counselling assistant, please answer the mental health questions based on the patient's description.
|
|
|
11 |
- medical
|
12 |
---
|
13 |
|
14 |
+
# Model Card for Model MentalChat-16K
|
15 |
|
16 |
## Model Details
|
17 |
|
|
|
70 |
device_map={"": 0}, token=""
|
71 |
)
|
72 |
|
73 |
+
model = PeftModel.from_pretrained(base_model,"Rustamshry/MentalChat-16K")
|
74 |
|
75 |
instruction = """
|
76 |
You are a helpful mental health counselling assistant, please answer the mental health questions based on the patient's description.
|