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 |
||||
typing_extensions==4.3.0 |
||||
min-dalle |
||||
|
@ -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