xnetba commited on
Commit
3847bfa
·
1 Parent(s): f82a3d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -9
app.py CHANGED
@@ -14,7 +14,7 @@ def get_usernames(model: str):
14
  Returns:
15
  (str, str, str, str): pre-prompt, username, bot name, separator
16
  """
17
- if model in ("OpenAssistant/oasst-sft-1-pythia-12b", "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5"):
18
  return "", "<|prompter|>", "<|assistant|>", "<|endoftext|>"
19
 
20
  def predict(
@@ -52,7 +52,7 @@ def predict(
52
 
53
  partial_words = ""
54
 
55
- if model in ("OpenAssistant/oasst-sft-1-pythia-12b", "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5"):
56
  iterator = client.generate_stream(
57
  total_inputs,
58
  typical_p=typical_p,
@@ -109,7 +109,7 @@ def radio_on_change(
109
  repetition_penalty,
110
  watermark,
111
  ):
112
- if value in ("OpenAssistant/oasst-sft-1-pythia-12b", "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5"):
113
  typical_p = typical_p.update(value=0.2, visible=True)
114
  top_p = top_p.update(visible=False)
115
  top_k = top_k.update(visible=False)
@@ -139,12 +139,6 @@ def radio_on_change(
139
  title = """<h1 align="center">LLM Chat</h1>"""
140
  description = """LLM Chat predložak:
141
  ```
142
- User: <utterance>
143
- Assistant: <utterance>
144
- User: <utterance>
145
- Assistant: <utterance>
146
- ...
147
- ```
148
  Mjenjajući predložak.
149
  """
150
 
 
14
  Returns:
15
  (str, str, str, str): pre-prompt, username, bot name, separator
16
  """
17
+ if model in ("OpenAssistant/oasst-sft-1-pythia-12b"):
18
  return "", "<|prompter|>", "<|assistant|>", "<|endoftext|>"
19
 
20
  def predict(
 
52
 
53
  partial_words = ""
54
 
55
+ if model in ("OpenAssistant/oasst-sft-1-pythia-12b"):
56
  iterator = client.generate_stream(
57
  total_inputs,
58
  typical_p=typical_p,
 
109
  repetition_penalty,
110
  watermark,
111
  ):
112
+ if value in ("OpenAssistant/oasst-sft-1-pythia-12b"):
113
  typical_p = typical_p.update(value=0.2, visible=True)
114
  top_p = top_p.update(visible=False)
115
  top_k = top_k.update(visible=False)
 
139
  title = """<h1 align="center">LLM Chat</h1>"""
140
  description = """LLM Chat predložak:
141
  ```
 
 
 
 
 
 
142
  Mjenjajući predložak.
143
  """
144