Update README.md

README.md

@@ -16,14 +16,16 @@ The `files/` folder contains the files needed in our code [here](https://snap-re
 - `files/pinverse.pt`
   - Precomputed pseudoinverse of 'proj.pt', used for obtaining the classifier weight space directions given labels.
 - `files/identity_df.pt`
-  - A pandas dataframe with each identity model in our dataset of weights and their associated binary labels. Used for getting labels for training linear classifiers. 
+  - A pandas dataframe with binary attribute labels for each identity-encoding model. Used for getting labels for training linear classifiers. 
 - `files/weight_dimensions.pt`
-  - A dictionary of the dimensionality for each LoRA module in the diffusion UNet. Used to save models in Diffusers pipeline format. 
+  - A dictionary of the dimensionality for each LoRA module in the diffusion UNet. Used to save models in Diffusers pipeline format.
+- `files/adapter_config.json`
+  - A configuration file for LoRA. Used to save models in Diffusers pipeline format. 
 
-**Drag-Based Manipulation**
-- `readout_sdxl_drag_correspondence.pt`
-- `readout_sdv15_drag_correspondence.pt`
-    - Readout head trained with a contrastive loss with [CoTracker](https://co-tracker.github.io) point tracks across pairs of [DAVIS](https://davischallenge.org) video frames.
-- `readout_sdxl_drag_appearance.pt`
-- `readout_sdv15_drag_appearance.pt`
-    - Readout head trained with a triplet loss with real frames as positives and [SDEdit-ed](https://sde-image-editing.github.io) frames as negatives derived from [DAVIS](https://davischallenge.org) videos.
+# Datasets of Model Weights
+The `weights_datasets`folder contains two datasets of model weights, one set containing LoRA weights encoding different human visual identities, and another for different dog breeds. We also provide metadata and files for handling these datasets.
+- `weights_datasets/identities/all_weights.pt`
+  - 64974x99648 dimensional tensor where each row is a 99648-dimensional vector of flattened LoRAs for each identity-encoding model.
+- `weights_datasets/identities/identity_df.pt`
+  - A pandas dataframe with binary attribute labels for each identity-encoding model.  
+