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_torch.py
+4 -4
View File
@@ -4,17 +4,17 @@ from torch import Tensor
import argparse
from typing import Dict
from image_from_text import (
from min_dalle.image_from_text import (
load_dalle_bart_metadata,
tokenize,
detokenize_torch,
save_image,
ascii_from_image
)
from models.dalle_bart_encoder_torch import DalleBartEncoderTorch
from models.dalle_bart_decoder_torch import DalleBartDecoderTorch
from min_dalle.models.dalle_bart_encoder_torch import DalleBartEncoderTorch
from min_dalle.models.dalle_bart_decoder_torch import DalleBartDecoderTorch
from load_params import (
from min_dalle.load_params import (
load_dalle_bart_flax_params,
convert_dalle_bart_torch_from_flax_params
)