Skip to content

Commit

Permalink
updates for pypi upload
Browse files Browse the repository at this point in the history
  • Loading branch information
jvendrow committed Feb 27, 2023
1 parent 5953d9e commit e4824f7
Show file tree
Hide file tree
Showing 3 changed files with 10 additions and 5 deletions.
5 changes: 4 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,10 @@ Blog post: [https://gradientscience.org/dataset-interfaces/](https://gradientsci
</p>

## Getting started
To begin, clone our repository.
Install using pip, or clone our repository.
```
pip install dataset-interfaces
```

**Example:** For a walkthrough of codebase, check out our [example notebook](notebooks/Example.ipynb). This notebook shows how to
construct a dataset interface for a subset of ImageNet and generate counterfactual examples.
Expand Down
4 changes: 3 additions & 1 deletion dataset_interfaces/__init__.py
Original file line number Diff line number Diff line change
@@ -1,2 +1,4 @@
__version__ = "0.1.0"

from .textual_inversion import run_textual_inversion
from .generate import generate
from .generate import generate
6 changes: 3 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@

setup(
name='dataset_interfaces',
version='0.0.1',
version='0.1.0',
description='Dataset Interfaces',
url='https://github.com/MadryLab/dataset-interfacees',
url='https://github.com/MadryLab/dataset-interfaces',
author='MadryLab',
author_email='jvendrow@mit.edu',
license = 'MIT',
Expand All @@ -22,7 +22,7 @@
'numpy',
'torchvision',
'diffusers',
'transformers'
'transformers',
'accelerate'
],
)

0 comments on commit e4824f7

Please sign in to comment.