Skip to content

fix: pass all arguments through gradient checkpoint in BasicTransformerBlock - #474

Closed
Mr-Neutr0n wants to merge 1 commit into
Stability-AI:mainfrom
Mr-Neutr0n:fix/checkpoint-pass-all-args
Closed

fix: pass all arguments through gradient checkpoint in BasicTransformerBlock#474
Mr-Neutr0n wants to merge 1 commit into
Stability-AI:mainfrom
Mr-Neutr0n:fix/checkpoint-pass-all-args