NeuralInternet's picture
Update run.py
425165e
import os
# os.system('python download-model.py PygmalionAI/pygmalion-350m --branch main')
os.system('python download-model.py EleutherAI/gpt-neo-1.3B --branch main')
# os.system('python download-model.py EleutherAI/gpt-neo-2.7B --branch main')
# os.system('python download-model.py waifu-workshop/pygmalion-6b --branch original-sharded')
# os.system('python server.py --cpu --cai-chat --model pygmalion-350m')
os.system('python server.py --model gpt-neo-1.3B')