This commit is contained in:
alexisxy 2023-09-23 00:16:36 -04:00
parent 741292e1d2
commit 1ee1ea4800

View File

@ -27,10 +27,10 @@ class PromptConstructor(object):
lm_config: lm_config.LMConfig, lm_config: lm_config.LMConfig,
tokenizer: Tokenizer, tokenizer: Tokenizer,
): ):
self.instrction_path = Path(instruction_path) self.instruction_path = Path(instruction_path)
self.obs_modality = "text" self.obs_modality = "text"
self.lm_config = lm_config self.lm_config = lm_config
instruction = json.load(open(self.instrction_path)) instruction = json.load(open(self.instruction_path))
instruction["examples"] = [tuple(e) for e in instruction["examples"]] instruction["examples"] = [tuple(e) for e in instruction["examples"]]
self.instruction: Instruction = instruction self.instruction: Instruction = instruction
self.tokenizer = tokenizer self.tokenizer = tokenizer