From 5d97147fd81616187e0148291af80ae0c79d863a Mon Sep 17 00:00:00 2001 From: StochasticRomanAgeev Date: Tue, 18 Apr 2023 20:10:36 +0800 Subject: [PATCH] fix: removed test file Removed file from tests file --- tests/xturing/engines/test_llama_int4.py | 21 --------------------- 1 file changed, 21 deletions(-) delete mode 100644 tests/xturing/engines/test_llama_int4.py diff --git a/tests/xturing/engines/test_llama_int4.py b/tests/xturing/engines/test_llama_int4.py deleted file mode 100644 index c4dc3ac..0000000 --- a/tests/xturing/engines/test_llama_int4.py +++ /dev/null @@ -1,21 +0,0 @@ -from xturing.models import LlamaLoraInt4 -from xturing.datasets import TextDataset - -DATASET_OTHER_EXAMPLE_DICT = { - "text": ["first text", "second text"], - "target": ["first text", "second text"], -} - - -model = LlamaLoraInt4("/home/romanageev/peft/llama7b-4bit-128g.pt") -dataset = TextDataset(DATASET_OTHER_EXAMPLE_DICT) -# finetuning_config = model.finetuning_config() -# finetuning_config.num_train_epochs = 1 -# model.finetune(dataset=dataset) -generation_config = model.generation_config() -generation_config.do_sample = False -generation_config.max_new_tokens = 20 -generation_config.top_k = 50 -generation_config.top_p = 1.0 -result = model.generate(dataset=dataset) -print(result) \ No newline at end of file