Spaces:

Locutusque
/

Locutusque-Models

Running on Zero

Locutusque commited on Feb 5, 2024

Commit

2963fd2

verified ·

1 Parent(s): f80709d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,6 @@ def generate(
     # Set tokenize correctly. Otherwise ticking the box breaks it.
     if model_name == "Locutusque/TinyMistral-248M-v3":
         prompt = user_input
-        gr.Interface.notify("Warning: Locutusque/TinyMistral-248M-v3 is an experimental preview model, performance is not guaranteed", type="warning")
     else:
         prompt = f"<|im_start|>user\n{user_input}<|im_end|>\n<|im_start|>assistant\n"
     outputs = pipe(prompt, max_new_tokens=max_new_tokens, do_sample=True,

     # Set tokenize correctly. Otherwise ticking the box breaks it.
     if model_name == "Locutusque/TinyMistral-248M-v3":
         prompt = user_input
     else:
         prompt = f"<|im_start|>user\n{user_input}<|im_end|>\n<|im_start|>assistant\n"
     outputs = pipe(prompt, max_new_tokens=max_new_tokens, do_sample=True,