Spaces:
Runtime error
Runtime error
Antoine Chaffin
commited on
Commit
•
87801f9
1
Parent(s):
1e2d20c
debug
Browse files
app.py
CHANGED
@@ -37,13 +37,16 @@ If a question does not make any sense, or is not factually coherent, explain why
|
|
37 |
"""
|
38 |
LEN_DEFAULT_PROMPT = len(tokenizer.encode(DEFAULT_SYSTEM_PROMPT))
|
39 |
def embed(user, max_length, window_size, method, prompt):
|
40 |
-
|
41 |
uid = USERS.index(user)
|
42 |
watermarker = Watermarker(tokenizer=tokenizer, model=model, window_size=window_size, payload_bits=PAYLOAD_BITS)
|
43 |
prompt = get_prompt(prompt)
|
|
|
44 |
watermarked_texts = watermarker.embed(key=args.key, messages=[ uid ],
|
45 |
max_length=max_length+LEN_DEFAULT_PROMPT, method=method, prompt=prompt)
|
46 |
print("===")
|
|
|
|
|
47 |
print(watermarked_texts[0].split("[/INST]")[1][0])
|
48 |
print("===")
|
49 |
print(watermarked_texts[0].split("[/INST]"))
|
|
|
37 |
"""
|
38 |
LEN_DEFAULT_PROMPT = len(tokenizer.encode(DEFAULT_SYSTEM_PROMPT))
|
39 |
def embed(user, max_length, window_size, method, prompt):
|
40 |
+
|
41 |
uid = USERS.index(user)
|
42 |
watermarker = Watermarker(tokenizer=tokenizer, model=model, window_size=window_size, payload_bits=PAYLOAD_BITS)
|
43 |
prompt = get_prompt(prompt)
|
44 |
+
print("prompt:", prompt)
|
45 |
watermarked_texts = watermarker.embed(key=args.key, messages=[ uid ],
|
46 |
max_length=max_length+LEN_DEFAULT_PROMPT, method=method, prompt=prompt)
|
47 |
print("===")
|
48 |
+
print(watermarked_text)
|
49 |
+
print("===")
|
50 |
print(watermarked_texts[0].split("[/INST]")[1][0])
|
51 |
print("===")
|
52 |
print(watermarked_texts[0].split("[/INST]"))
|