Update utils.py
Browse files
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 "
|
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")
|