Update README.md

README.md

@@ -1,10 +1,10 @@
 ---
 {}
 ---
-This repository contains the dataset of model weights and utility files for [Interpreting the Weight Space of Customized Diffusion Models](https://snap-research.github.io/weights2weights/) (aka *weights2weights*). The code can be found at https://github.com/snap-research/weights2weights.
+This repository contains the dataset of model weights and utility files for [Interpreting the Weight Space of Customized Diffusion Models](https://snap-research.github.io/weights2weights/) (aka *weights2weights*). The code can be found at https://github.com/snap-research/weights2weights. The `files` folder contains files that our code requires. The `weights_datasets` folder contains the datasets of model weights in addition to metadata and supplemental files.
 
 # Files for creating *weights2weights* Space
-The `files/` folder contains the files needed in our code [here](https://snap-research.github.io/weights2weights/) in order to conduct sampling, inversion, and editing in *weights2weights* space.
+The `files/` folder contains the files needed for our code [here](https://snap-research.github.io/weights2weights/) in order to conduct sampling, inversion, and editing in *weights2weights* space.
 - `files/V.pt`
   - 99648x10000 dimensional tensor used to project or unproject LoRA weights onto a principal component representation in *w2w* space or to unproject back into the LoRA space.
 - `files/mean.pt`
@@ -28,4 +28,11 @@ The `weights_datasets`folder contains two datasets of model weights, one set con
   - 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.  
- 
+- `weights_datasets/dogs/dog_weights.pt`
+  - 120x99648 dimensional tensor where each row is a 99648-dimensional vector of flattened LoRAs for each dog-encoding model.
+- `weights_datasets/dogs/dog_labels.pt`
+  - A list of the dog breeds corresponding to each entry in `dog_weights.pt`.
+- `weights_datasets/weight_dimensions.pt`
+  - A dictionary of the dimensionality for each LoRA module in the diffusion UNet. Used to save models in Diffusers pipeline format.
+- `weights_datasets/adapter_config.json`
+  - A configuration file for LoRA. Used to save models in Diffusers pipeline format.