Update README.md

README.md

@@ -29,15 +29,31 @@ PsychAdapter is a modular framework designed to adapt Large Language Models (LLM
 - **Repository:** https://github.com/humanlab/psychadapter
 - **Paper:** https://www.nature.com/articles/s44387-026-00071-9
 
-## Checkpoint Structure
+## Checkpoint Structure & Setup
 
-The provided checkpoints include the following components:
+The model checkpoints are provided as separate ZIP files to keep the psychological dimensions modular. Note that each model requires the shared decoder weights to function.
 
-1. **Base Models:** The foundational pre-trained LLM weights (e.g., Gemma-2b) used as the backbone for adaptation.
-2. **LoRA Weights (Big 5):** Specialized Low-Rank Adaptation (LoRA) weights trained specifically on the Big Five personality traits (Openness, Conscientiousness, Extraversion, Agreeableness, Neuroticism).
-3. **LoRA Weights (Mental Health):** Specialized LoRA weights trained to reflect mental health variables, specifically Depression and Satisfaction with Life (SWL).
+### 1. Downloaded Files:
+* **big5_model.zip**: LoRA weights and projection matrix weights for Big Five personality traits.
+* **dep_model.zip**: LoRA weights and projection matrix weights  for Depression markers.
+* **swl_model.zip**: LoRA weights and projection matrix weights  for Satisfaction with Life.
+* **decoder.zip**: Contains the shared base model/decoder weights required by all adapters.
+
+### 2. Assembly Instructions:
+All model directories contain an empty `decoder/` subdirectory. To use the models, you must manually populate these directories:
+
+1. Unzip your desired model (e.g., `big5_model.zip`).
+2. Unzip `decoder.zip`.
+3. Copy the entire **content** of the unzipped `decoder` folder into the `decoder/` directory located inside your model folder (e.g., `big5_model/decoder/`).
+
+**Note:** The models will fail to load if the `decoder/` directory is empty or missing the shared weights.
+
+## Checkpoint Details
+
+* **Base Models:** Found in `decoder.zip` (e.g., Gemma-2b backbone).
+* **LoRA Weights (Big 5):** Found in `big5_model.zip`, trained on Openness, Conscientiousness, Extraversion, Agreeableness, and Neuroticism.
+* **LoRA Weights (Mental Health):** Found in `dep_model.zip` (Depression) and `swl_model.zip` (Satisfaction with Life).
 
-These adapters are designed to be swapped or combined to steer the base model's output according to the desired psychological profile.
 
 ## Intended Use