ysharma HF Staff commited on
Commit
7e230a2
·
1 Parent(s): 3178c9a
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,7 +5,8 @@ text_to_text_gm130b = gr.Blocks.load(name="spaces/THUDM/GLM-130B")
5
 
6
  def block_inference(prompt):
7
  #thudm glm 130 b space
8
- generated_text = text_to_text_gm130b(model_input=prompt, seed=1234, out_seq_length=256, min_gen_length=0, sampling_strategy='BeamSearchStrategy', num_beams=2, length_penalty=1, no_repeat_ngram_size=3, temperature=0.7, topk=1, topp=0)
 
9
  return generated_text
10
 
11
  def chat(message, history):
 
5
 
6
  def block_inference(prompt):
7
  #thudm glm 130 b space
8
+ #generated_text = text_to_text_gm130b(model_input=prompt, seed=1234, out_seq_length=256, min_gen_length=0, sampling_strategy='BeamSearchStrategy', num_beams=2, length_penalty=1, no_repeat_ngram_size=3, temperature=0.7, topk=1, topp=0)
9
+ generated_text = text_to_text_gm130b(prompt, 1234, 256, 0, 'BeamSearchStrategy', 2, 1, 3, 0.7, 1, 0)
10
  return generated_text
11
 
12
  def chat(message, history):