Remove generate/lora (#1115)

Lightning-AI · Mar 15, 2024 · f13102d · f13102d
1 parent cede554
commit f13102d
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 206 deletions.
diff --git a/litgpt/__main__.py b/litgpt/__main__.py
@@ -13,7 +13,6 @@
 from litgpt.generate.adapter_v2 import main as generate_adapter_v2_fn
 from litgpt.generate.base import main as generate_base_fn
 from litgpt.generate.full import main as generate_full_fn
-from litgpt.generate.lora import main as generate_lora_fn
 from litgpt.generate.sequentially import main as generate_sequentially_fn
 from litgpt.generate.tp import main as generate_tp_fn
 from litgpt.pretrain import setup as pretrain_fn
@@ -55,7 +54,6 @@ def main() -> None:
             "help": "Generate text samples based on a model and tokenizer.",
             "base": {"fn": generate_base_fn, "help": "Default generation option."},
             "full": {"fn": generate_full_fn, "help": "For models finetuned with `litgpt finetune full`."},
-            "lora": {"fn": generate_lora_fn, "help": "For models finetuned with `litgpt finetune lora`."},
             "adapter": {"fn": generate_adapter_fn, "help": "For models finetuned with `litgpt finetune adapter`."},
             "adapter_v2": {
                 "fn": generate_adapter_v2_fn,

diff --git a/litgpt/generate/lora.py b/litgpt/generate/lora.py
diff --git a/tests/test_generate_lora.py b/tests/test_generate_lora.py