You need to install flax 0.4.2. If you're using collab you just open a terminal ... | Hacker News

Hacker Newsnew | past | comments | ask | show | jobs | submit

		jcims on June 28, 2022 \| parent \| context \| favorite \| on: Show HN: I stripped DALL·E Mini to its bare essent... You need to install flax 0.4.2. If you're using collab you just open a terminal (icon in the bottom left of the screen) and run: `pip3 install flax==0.4.2`

kuprel on June 28, 2022 [–]

Yes this is the fix for now. I need to address what is actually causing the dtype mismatch

Guidelines | FAQ | Lists | API | Security | Legal | Apply to YC | Contact