ZekunXi commited on
Commit
5f7f363
1 Parent(s): 4d61057

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -14,7 +14,7 @@ def edit(prompt, target_new):
14
  tok.pad_token_id = tok.eos_token_id
15
  global edit_model
16
  edit_model,_ = apply_grace_to_model(model,tok,request,hparams,keep_original_weight=True)
17
- return "finish"
18
 
19
  def generate(input_text):
20
  tok = GPT2Tokenizer.from_pretrained("./models/gpt2-xl")
 
14
  tok.pad_token_id = tok.eos_token_id
15
  global edit_model
16
  edit_model,_ = apply_grace_to_model(model,tok,request,hparams,keep_original_weight=True)
17
+ return "Knowledge editing has been completed. You can proceed with testing on the right."
18
 
19
  def generate(input_text):
20
  tok = GPT2Tokenizer.from_pretrained("./models/gpt2-xl")