Add use_flash_attention_2=True

iwiwi / epochraft-hf-fsdp

Example of using Epochraft to train HuggingFace transformers models with PyTorch FSDP

MIT License

12 stars 5 forks source link

Closed iwiwi closed 10 months ago