Skip to content

Palpitate18/ULMRec

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ULMRec

Step. 1

Change the configuration of parameters in config.py based on the position of the model

parser.add_argument('--llm_base_model', type=str, default='./llama-2-7b-hf') #need change
parser.add_argument('--llm_base_tokenizer', type=str, default='./llama-2-7b-hf') #need change

Step. 2

  • Install requirements.

Step. 3

  • Run cp_test.py to load checkpoint model.
python cp_test.py --dataset_code games
python cp_test.py --dataset_code beauty

Before run the file, it's needed to change some configs in config.py checkpoint_path for Beauty and Games should be replaced with an abosulate path instead of relative path.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages