gpt-omni commited on
Commit
fdfd286
·
1 Parent(s): 8978fcf
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -38,7 +38,7 @@ from litgpt.generate.base import sample
38
 
39
  device = "cuda" if torch.cuda.is_available() else "cpu"
40
  ckpt_dir = "./checkpoint"
41
- streaming_output = False
42
 
43
 
44
  OUT_CHUNK = 4096
@@ -335,5 +335,7 @@ demo = gr.Interface(
335
  # live=True,
336
  )
337
 
338
- demo.queue()
339
- demo.launch()
 
 
 
38
 
39
  device = "cuda" if torch.cuda.is_available() else "cpu"
40
  ckpt_dir = "./checkpoint"
41
+ streaming_output = True
42
 
43
 
44
  OUT_CHUNK = 4096
 
335
  # live=True,
336
  )
337
 
338
+
339
+ if __name__ == '__main__':
340
+ demo.queue()
341
+ demo.launch()