Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
avdravid authored Jul 19, 2024
1 parent 42c5e4a commit 782e201
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -20,9 +20,10 @@ $ pip install -r requirements.txt
```

### Files
The files needed to create *w2w* space, load models, train classifiers, etc. can be downloaded at this [link](https://drive.google.com/file/d/1W1_klpdeCZr5b0Kdp7SaS7veDV2ZzfbB/view?usp=sharing). Keep the folder structure and place it into the `weights2weights` folder containing all the code.
The Hugging Face model card is available at this [link] (https://huggingface.co/Snapchat/weights2weights). You can find info on all the files there.
The files needed to create *w2w* space, load models, train classifiers, etc. can be downloaded at this [link](https://drive.google.com/file/d/1W1_klpdeCZr5b0Kdp7SaS7veDV2ZzfbB/view?usp=sharing), or this [link](https://huggingface.co/Snapchat/weights2weights/tree/main/files). Keep the folder structure and place the `files` folder into the `weights2weights` folder containing all the code.

The dataset of full model weights (i.e. the full Dreambooth LoRA parameters) will be released within the next week (by June 21).
The dataset of full model weights (i.e. the full Dreambooth LoRA parameters) is available [here] (https://huggingface.co/Snapchat/weights2weights/tree/main/weights_datasets).

## Sampling
We provide an interactive notebook for sampling new identity-encoding models from *w2w* space in `sampling/sampling.ipynb`. Instructions are provided in the notebook. Once a model is sampled, you can run typical inference with various text prompts and generation seeds as with a typical personalized model.
Expand Down

0 comments on commit 782e201

Please sign in to comment.