tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fixed relative imports

Browse Source
This commit is contained in:
Brett Kuprel
2022-06-27 12:43:47 -04:00
parent 9cf9d46dd7
commit 018414a5c3
12 changed files with 137 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
image_from_text_flax.py
+4 -4
View File
@@ -3,16 +3,16 @@ from jax import numpy as jnp
import numpy
import argparse
from load_params import load_dalle_bart_flax_params
from image_from_text import (
from min_dalle.load_params import load_dalle_bart_flax_params
from min_dalle.image_from_text import (
load_dalle_bart_metadata,
tokenize,
detokenize_torch,
save_image,
ascii_from_image
)
from models.dalle_bart_encoder_flax import DalleBartEncoderFlax
from models.dalle_bart_decoder_flax import DalleBartDecoderFlax
from min_dalle.models.dalle_bart_encoder_flax import DalleBartEncoderFlax
from min_dalle.models.dalle_bart_decoder_flax import DalleBartDecoderFlax
parser = argparse.ArgumentParser()