parent
f0c8f258e9
commit
be2beca7c0
9 changed files with 116 additions and 58 deletions
File diff suppressed because one or more lines are too long
@ -0,0 +1,3 @@ |
|||||||
|
from .dalle_bart_encoder_torch import DalleBartEncoderTorch |
||||||
|
from .dalle_bart_decoder_torch import DalleBartDecoderTorch |
||||||
|
from .vqgan_detokenizer import VQGanDetokenizer |
@ -1,2 +1 @@ |
|||||||
torch==1.12.0 |
min-dalle |
||||||
typing_extensions==4.3.0 |
|
||||||
|
@ -0,0 +1,23 @@ |
|||||||
|
import setuptools |
||||||
|
|
||||||
|
setuptools.setup( |
||||||
|
name='min-dalle', |
||||||
|
description = 'min(DALL·E)', |
||||||
|
version='0.1.4', |
||||||
|
author='Brett Kuprel', |
||||||
|
author_email = 'brkuprel@gmail.com', |
||||||
|
packages=[ |
||||||
|
'min_dalle', |
||||||
|
'min_dalle.models' |
||||||
|
], |
||||||
|
license='MIT', |
||||||
|
install_requires=[ |
||||||
|
'torch>=1.11.0', |
||||||
|
'typing_extensions>=4.1.0' |
||||||
|
], |
||||||
|
keywords = [ |
||||||
|
'artificial intelligence', |
||||||
|
'deep learning', |
||||||
|
'text to image' |
||||||
|
] |
||||||
|
) |
@ -1,25 +0,0 @@ |
|||||||
#!/bin/bash |
|
||||||
|
|
||||||
set -e |
|
||||||
|
|
||||||
pip3 install -r requirements.txt |
|
||||||
|
|
||||||
repo_path="https://huggingface.co/kuprel/min-dalle/resolve/main" |
|
||||||
|
|
||||||
mini_path="./pretrained/dalle_bart_mini" |
|
||||||
mega_path="./pretrained/dalle_bart_mega" |
|
||||||
vqgan_path="./pretrained/vqgan" |
|
||||||
|
|
||||||
mkdir -p ${vqgan_path} |
|
||||||
mkdir -p ${mini_path} |
|
||||||
mkdir -p ${mega_path} |
|
||||||
|
|
||||||
curl ${repo_path}/detoker.pt -L --output ${vqgan_path}/detoker.pt |
|
||||||
curl ${repo_path}/vocab_mini.json -L --output ${mini_path}/vocab.json |
|
||||||
curl ${repo_path}/merges_mini.txt -L --output ${mini_path}/merges.txt |
|
||||||
curl ${repo_path}/encoder_mini.pt -L --output ${mini_path}/encoder.pt |
|
||||||
curl ${repo_path}/decoder_mini.pt -L --output ${mini_path}/decoder.pt |
|
||||||
curl ${repo_path}/vocab.json -L --output ${mega_path}/vocab.json |
|
||||||
curl ${repo_path}/merges.txt -L --output ${mega_path}/merges.txt |
|
||||||
curl ${repo_path}/encoder.pt -L --output ${mega_path}/encoder.pt |
|
||||||
curl ${repo_path}/decoder.pt -L --output ${mega_path}/decoder.pt |
|
Loading…
Reference in new issue