Merge pull request #109 from huggingface/smangrul/add-gpt-neox

add `EleutherAI/gpt-neox-20b` to support matrix
This commit is contained in:
Sourab Mangrulkar
2023-02-18 12:38:04 +05:30
committed by GitHub
+3
View File
@@ -222,6 +222,7 @@ An example is provided in `~examples/causal_language_modeling/peft_lora_clm_acce
| OPT | ✅ | ✅ | ✅ | ✅ |
| GPT-Neo | ✅ | ✅ | ✅ | ✅ |
| GPT-J | ✅ | ✅ | ✅ | ✅ |
| GPT-NeoX-20B | ✅ | ✅ | ✅ | ✅ |
### Conditional Generation
| Model | LoRA | Prefix Tuning | P-Tuning | Prompt Tuning |
@@ -279,6 +280,8 @@ The same principle applies to our [segmentation models](https://huggingface.co/m
| Model | LoRA | Prefix Tuning | P-Tuning | Prompt Tuning |
| --------- | ---- | ---- | ---- | ---- |
| SegFormer | ✅ | | | |
## Caveats:
1. Below is an example of using PyTorch FSDP for training. However, it doesn't lead to