omkar56 commited on
Commit
a1daa46
·
1 Parent(s): 4759ffb

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -40,9 +40,9 @@ async def generate_text(request: Request, prompt: str = Body()):
40
  repetition_penalty=repetition_penalty,
41
  do_sample=True,
42
  seed=random.randint(0, 10**7),
43
- stream=True,
44
  details=True,
45
- return_full_text=False
46
  )
47
  # output = ""
48
 
 
40
  repetition_penalty=repetition_penalty,
41
  do_sample=True,
42
  seed=random.randint(0, 10**7),
43
+ stream=False,
44
  details=True,
45
+ return_full_text=True
46
  )
47
  # output = ""
48