I couldn't get the "mega" option to work, I got an error "TypeError: lax.dynamic_update_slice requires arguments to have the same dtypes, got float32, float16" (looks like a known issue https://github.com/kuprel/min-dalle/issues/2)
Edit: installing flax 0.4.2 fixes this issue, thank all!
Macbook Pro M1 Pro numbers (CPU):
python3 image_from_text.py --text='court sketch of godzilla on trial' --mega 640.24s user 179.30s system 544% cpu 2:30.39 totalpython3 image_from_text.py --text='a comfy chair that looks like an avocado' 612.30s user 180.72s system 552% cpu 2:23.52 total
So much for semver :(
> Major version zero (0.y.z) is for initial development. Anything MAY change at any time. The public API SHOULD NOT be considered stable.
Variant schemes like the one respected by Cargo (https://doc.rust-lang.org/cargo/reference/semver.html) aren't usually much different.
> Initial development releases starting with "0.y.z" can treat changes in "y" as a major release, and "z" as a minor release.
https://github.com/kuprel/min-dalle/commit/38ebe54a382f36dc7...