lonestar108 commited on
Commit
256d625
·
1 Parent(s): 7d3fdd3
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -1,3 +1,3 @@
1
  import os
2
  os.system('python download-model.py PygmalionAI/pygmalion-350m --branch main')
3
- os.system('python server.py --cpu --auto-devices')
 
1
  import os
2
  os.system('python download-model.py PygmalionAI/pygmalion-350m --branch main')
3
+ os.system('python server.py --auto-devices --gpu-memory 24 --public-api')