Consider depending on dalle pytorch pypi module

tgisaturday / dalle-lightning

Refactoring dalle-pytorch and taming-transformers for TPU VM

60 stars 12 forks source link

Consider depending on dalle pytorch pypi module #2

Open rom1504 opened 3 years ago

rom1504 commented 3 years ago

I think that would help avoid some code duplication Dalle pytorch repo is quite open to PRs

tgisaturday commented 3 years ago

@rom1504 It would be convenient but right now I'm having trouble with third-party libs like einops on TPUs. I will consider it after main code structure is fixed.

tgisaturday commented 3 years ago

@rom1504 I'm going to change current dalle model to fully import from DALLE-pytorch and create TPU optimized dalle model with Net2Net cond-transformer concept from taming repo.