jwkirchenbauer commited on
Commit
2c928af
1 Parent(s): e65e5ab
Files changed (1) hide show
  1. demo_watermark.py +1 -1
demo_watermark.py CHANGED
@@ -362,7 +362,7 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
362
  Longer prompts and stopping mid sentence often helps encourage more fluent, longer genrations.
363
  """
364
  )
365
- gr.Markdown(f"Language model: {args.model_name_or_path}")
366
 
367
  # Construct state for parameters, define updates and toggles
368
  default_prompt = args.__dict__.pop("default_prompt")
 
362
  Longer prompts and stopping mid sentence often helps encourage more fluent, longer genrations.
363
  """
364
  )
365
+ gr.Markdown(f"Language model: {args.model_name_or_path} {'(float16 mode)' if args.load_fp16 else ''}")
366
 
367
  # Construct state for parameters, define updates and toggles
368
  default_prompt = args.__dict__.pop("default_prompt")