admin/LLM-from-scratch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add dolly test

Browse Source
This commit is contained in:
coffee
2025-05-09 17:47:11 +08:00
parent 411124cfca
commit 95f84f0c4c
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/temp_test.py
+4 -5
View File
@@ -5,12 +5,11 @@ from torch import nn
from src_llm.configuration_dolly import DollyConfig from src_llm.configuration_dolly import DollyConfig
from src_llm.modeling_dolly import DollyModel from src_llm.modeling_dolly import DollyModel
# model = DollyModel(DollyConfig()) # model = DollyModel(DollyConfig())
# model.save_pretrained("./my_model") # model.save_pretrained("./dolly_model")
# model.config.save_pretrained("./my_model") # model.config.save_pretrained("./dolly_model")
config = DollyConfig.from_pretrained("./my_model")
model = DollyModel.from_pretrained("./my_model", config=config)
config = DollyConfig.from_pretrained("./dolly_model")
model = DollyModel.from_pretrained("./dolly_model", config=config)
# input_ids = torch.tensor([[101, 1024, 102]]).long() # input_ids = torch.tensor([[101, 1024, 102]]).long()
# outputs = model(input_ids) # outputs = model(input_ids)