Skip to content

Commit e2111af

Browse files
awaelchlirasbt
authored andcommitted
Remove generate/lora (#1115)
1 parent 09f2406 commit e2111af

File tree

3 files changed

+0
-206
lines changed

3 files changed

+0
-206
lines changed

litgpt/__main__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
from litgpt.generate.adapter_v2 import main as generate_adapter_v2_fn
1414
from litgpt.generate.base import main as generate_base_fn
1515
from litgpt.generate.full import main as generate_full_fn
16-
from litgpt.generate.lora import main as generate_lora_fn
1716
from litgpt.generate.sequentially import main as generate_sequentially_fn
1817
from litgpt.generate.tp import main as generate_tp_fn
1918
from litgpt.pretrain import setup as pretrain_fn
@@ -55,7 +54,6 @@ def main() -> None:
5554
"help": "Generate text samples based on a model and tokenizer.",
5655
"base": {"fn": generate_base_fn, "help": "Default generation option."},
5756
"full": {"fn": generate_full_fn, "help": "For models finetuned with `litgpt finetune full`."},
58-
"lora": {"fn": generate_lora_fn, "help": "For models finetuned with `litgpt finetune lora`."},
5957
"adapter": {"fn": generate_adapter_fn, "help": "For models finetuned with `litgpt finetune adapter`."},
6058
"adapter_v2": {
6159
"fn": generate_adapter_v2_fn,

litgpt/generate/lora.py

Lines changed: 0 additions & 136 deletions
This file was deleted.

tests/test_generate_lora.py

Lines changed: 0 additions & 68 deletions
This file was deleted.

0 commit comments

Comments
 (0)