Skip to content

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

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

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

fix: pass additional_tokens and n_times_crossframe_attn_in_self throu…

4659f8e
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs