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")
|