up model

app.py

@@ -23,7 +23,8 @@ arg_dict = {
     # 'model_name_or_path': 'facebook/opt-125m', 
     # 'model_name_or_path': 'facebook/opt-1.3b', 
     # 'model_name_or_path': 'facebook/opt-2.7b', 
-    'model_name_or_path': 'facebook/opt-6.7b',
+    # 'model_name_or_path': 'facebook/opt-6.7b',
+    'model_name_or_path': 'facebook/opt-13b',
     'load_fp16' : True,
     # 'load_fp16' : False,
     'prompt_max_length': None, 

demo_watermark.py

@@ -328,10 +328,7 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
         gr.Markdown("## 💧 [A Watermark for Large Language Models](https://arxiv.org/abs/2301.10226) 🔍")
         with gr.Row():
             gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=tomg-group-umd_lm-watermarking)")
-        with gr.Row():
             gr.Markdown("[jwkirchenbauer/lm-watermarking![](https://badgen.net/badge/icon/GitHub?icon=github&label)](https://github.com/jwkirchenbauer/lm-watermarking)")
-        with gr.Row():
-            gr.Markdown(f"Language model: {args.model_name_or_path}")
         with gr.Accordion("Understanding the output metrics",open=False):
             gr.Markdown(
             """
@@ -365,6 +362,7 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
                 Longer prompts and stopping mid sentence often helps encourage more fluent, longer genrations.
                 """
                 )
+        gr.Markdown(f"Language model: {args.model_name_or_path}")
 
         # Construct state for parameters, define updates and toggles
         default_prompt = args.__dict__.pop("default_prompt")