Set use_memory_efficient_attention to False and modify config.json of 2_Dense_8192 1bd9d2a bglearning commited on Sep 30