dhuynh95 commited on
Commit
cc06a7b
1 Parent(s): b9ce19a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -393,11 +393,12 @@ with gr.Blocks(theme=style) as demo:
393
  You can reduce the Prefix size in the Advanced parameters to reduce the context length and see if the model still extracts the training sample.""")
394
  with gr.Row():
395
  with gr.Column():
396
- with gr.Accordion("Prefix size", open=True):
397
- k = gr.Slider(minimum=1, maximum=DEFAULT_SAMPLE_MAX_TOKENS, value=DEFAULT_K,
398
- step=1,
399
- info="""Number of tokens used to assess memorization.
400
- The more tokens are used, the more likely one can observe the LLM finishing the prompt with the verbatim code used in the training set.""")
 
401
  submit = gr.Button("Check memorization", variant="primary")
402
  examples_dropdown = gr.Dropdown(choices=list(examples.keys()), value=list(examples.keys())[0],
403
  interactive=True,
 
393
  You can reduce the Prefix size in the Advanced parameters to reduce the context length and see if the model still extracts the training sample.""")
394
  with gr.Row():
395
  with gr.Column():
396
+
397
+ k = gr.Slider(minimum=1, maximum=DEFAULT_SAMPLE_MAX_TOKENS, value=DEFAULT_K,
398
+ step=1,
399
+ label="Prefix size",
400
+ info="""Number of tokens we keep from the original sample to see if the LLM will complete the prompt with the rest of the training sample.
401
+ The more tokens are used, the more likely one can observe the LLM finishing the prompt with the verbatim code used in the training set.""")
402
  submit = gr.Button("Check memorization", variant="primary")
403
  examples_dropdown = gr.Dropdown(choices=list(examples.keys()), value=list(examples.keys())[0],
404
  interactive=True,