jslin09 commited on
Commit
a314573
·
verified ·
1 Parent(s): 62e026c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def rnd_generate(prompt):
19
  set_seed(rnd_seed)
20
  inputs = tokenizer(prompt, return_tensors="pt").to('cuda') # 回傳的張量使用 Pytorch的格式。如果是 Tensorflow 格式的話,則指定為 "tf"。
21
  results = model.generate(inputs["input_ids"],
22
- max_length=500,
23
  num_return_sequences=1, # 產生 1 個句子回來。
24
  do_sample=True,
25
  temperature=0.75,
 
19
  set_seed(rnd_seed)
20
  inputs = tokenizer(prompt, return_tensors="pt").to('cuda') # 回傳的張量使用 Pytorch的格式。如果是 Tensorflow 格式的話,則指定為 "tf"。
21
  results = model.generate(inputs["input_ids"],
22
+ max_length=600,
23
  num_return_sequences=1, # 產生 1 個句子回來。
24
  do_sample=True,
25
  temperature=0.75,