1580158
/
miaoshouai-assistant
mirror of https://github.com/miaoshouai/miaoshouai-assistant.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update install.py

pull/12/head
Superag 2023-04-24 16:05:49 -07:00
parent 2148ef6c56
commit 7620c729fa
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
install.py
View File

@ -2,8 +2,6 @@ import launch
import os import os
import gzip import gzip
import io import io
from scripts.runtime.msai_prelude import MiaoshouPrelude
def install_preset_models_if_needed(): def install_preset_models_if_needed():
assets_folder = os.path.join(os.path.dirname(os.path.realpath(__file__)), "assets") assets_folder = os.path.join(os.path.dirname(os.path.realpath(__file__)), "assets")
@ -20,9 +18,13 @@ def install_preset_models_if_needed():
model_file.write(content) model_file.write(content)
def create_folders(): def create_folders():
prelude = MiaoshouPrelude() cover_folder = os.path.join(os.path.dirname(os.path.realpath(__file__)), "cover")
os.mkdir(prelude.cover_folder) cache_folder = os.path.join(os.path.dirname(os.path.realpath(__file__)), "cache")
os.mkdir(prelude.cache_folder)
if not os.path.exists(cover_folder):
os.mkdir('cover')
if not os.path.exists(cache_folder):
os.mkdir('cache')
req_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), "requirements.txt") req_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), "requirements.txt")