Update to StableLM code (#514)

* StableLM now part of Transformers as stablelm rather than stablelm_epoch; changed config to match new changes

* removing old file

* reference new stablelm
This commit is contained in:
Ashish
2024-03-01 10:53:38 -07:00
committed by GitHub
parent 3acc1ec84e
commit 261f1280f6
2 changed files with 9 additions and 9 deletions

View File

@@ -29,7 +29,7 @@ def linear_to_lora_layers(model: nn.Module, num_lora_layers: int):
"llama",
"phi",
"mixtral",
"stablelm_epoch",
"stablelm",
"qwen2",
"gemma",
]: