From fa7e627cf525cf9a61e5cdeb7bc8c02297cc037e Mon Sep 17 00:00:00 2001 From: coffee Date: Fri, 9 May 2025 17:47:11 +0800 Subject: [PATCH] add dolly test --- test/temp_test.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/test/temp_test.py b/test/temp_test.py index 0d7ba88..24617e0 100755 --- a/test/temp_test.py +++ b/test/temp_test.py @@ -5,12 +5,11 @@ from torch import nn from src_llm.configuration_dolly import DollyConfig from src_llm.modeling_dolly import DollyModel # model = DollyModel(DollyConfig()) -# model.save_pretrained("./my_model") -# model.config.save_pretrained("./my_model") - -config = DollyConfig.from_pretrained("./my_model") -model = DollyModel.from_pretrained("./my_model", config=config) +# model.save_pretrained("./dolly_model") +# model.config.save_pretrained("./dolly_model") +config = DollyConfig.from_pretrained("./dolly_model") +model = DollyModel.from_pretrained("./dolly_model", config=config) # input_ids = torch.tensor([[101, 1024, 102]]).long() # outputs = model(input_ids)