Update README.md

README.md

@@ -9,7 +9,8 @@ tags:
 
 <img src="teaser.jpg" alt="teaser" width="800"/>
 
-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 auxiliary files that our code requires to run. The `weights_datasets` folder contains the datasets of model weights in addition to metadata and supplemental files.
+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 auxiliary files that our code requires to run. The `weights_datasets` folder contains the datasets of model weights in addition to metadata and supplemental files. If you are interested in the dataset of images used to create our dataset of model weights, visit this [link] (https://huggingface.co/datasets/wangkua1/w2w-celeba-generated/tree/main). 
+
 
 # Files for creating *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.
@@ -43,4 +44,6 @@ The `weights_datasets` folder contains two datasets of model weights, one set co
 - `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.
+  - A configuration file for LoRA. Used to save models in Diffusers pipeline format.
+# Recreating the Dataset of Weights
+The images used to conduct Dreambooth fine-tuning can be found at this [link](https://huggingface.co/datasets/wangkua1/w2w-celeba-generated/tree/main). We provide instructions for how to recreate the dataset in our [GitHub Repo](https://github.com/snap-research/weights2weights/tree/main).