MrD05 commited on
Commit
4769593
1 Parent(s): 8e85ae6

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -1,4 +1,4 @@
1
  import os
2
  os.system('python download-model.py PygmalionAI/pygmalion-350m --branch main')
3
  # os.system('python download-model.py waifu-workshop/pygmalion-6b --branch original-sharded')
4
- os.system('python server.py --cpu --chat --model pygmalion-350m --no-stream')
 
1
  import os
2
  os.system('python download-model.py PygmalionAI/pygmalion-350m --branch main')
3
  # os.system('python download-model.py waifu-workshop/pygmalion-6b --branch original-sharded')
4
+ os.system('python server.py --cpu --chat --model pygmalion-350m --no-stream --auto-devices')