Spaces:
Sleeping
Sleeping
Commit
·
2c928af
1
Parent(s):
e65e5ab
up model
Browse files- 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")
|