Szeyu commited on
Commit
c98a781
·
verified ·
1 Parent(s): 6f079c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -40,9 +40,11 @@ def generate_content(image):
40
  raw = storyer(
41
  prompt,
42
  max_new_tokens=150,
43
- temperature=0.7,
44
- top_p=0.9,
45
- no_repeat_ngram_size=2,
 
 
46
  return_full_text=False
47
  )[0]["generated_text"].strip()
48
 
 
40
  raw = storyer(
41
  prompt,
42
  max_new_tokens=150,
43
+ temperature=0.8,
44
+ top_p=0.85,
45
+ top_k=50,
46
+ no_repeat_ngram_size=3,
47
+ repetition_penalty=1.15,
48
  return_full_text=False
49
  )[0]["generated_text"].strip()
50