antonovmaxim commited on
Commit
be23485
1 Parent(s): 35a137f

Update text-generation-webui/server.py

Browse files
Files changed (1) hide show
  1. text-generation-webui/server.py +2 -1
text-generation-webui/server.py CHANGED
@@ -6,7 +6,8 @@ import sys
6
  import time
7
  import zipfile
8
  import os
9
- os.system('cd text-generation-webui && python download-model.py waifu-workshop/pygmalion-6b --branch original-sharded')
 
10
  from pathlib import Path
11
 
12
  import gradio as gr
 
6
  import time
7
  import zipfile
8
  import os
9
+ os.system('cd text-generation-webui && python download-model.py PygmalionAI/pygmalion-350m --branch main')
10
+ # os.system('cd text-generation-webui && python download-model.py waifu-workshop/pygmalion-6b --branch original-sharded')
11
  from pathlib import Path
12
 
13
  import gradio as gr