textgendemo / run.py
justkorudev's picture
Update run.py
c87b966
raw
history blame
236 Bytes
import os
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 --model gpt-neo-2.7B')