Fix DiscPool dtype mismatch for autocast outputs#3

benbennett wants to merge 1 commit intomainbenbennett/SIMPLE:mainfrom

codex/fix-dtype-mismatch-in-image-pool-insertbenbennett/SIMPLE:codex/fix-dtype-mismatch-in-image-pool-insertCopy head branch name to clipboard