ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-04-10 12:19:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-03-05 00:14:15 +01:00
parent b643ad2860
commit 2496b9c4dc
2 changed files with 27 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
lollms/services/sd/lollms_sd.py
View File

@ -70,6 +70,21 @@ def install_sd(lollms_app:LollmsApplication):
subprocess.run(["git", "clone", "https://github.com/ParisNeo/SD-CN-Animation.git", str(sd_folder/"extensions/SD-CN-Animation")])
if show_yes_no_dialog("warning!","Do you want to install a model from civitai?\nIsuggest dreamshaper xl."):
download_file("https://civitai.com/api/download/models/351306", sd_folder/"models/Stable-diffusion","dreamshaperXL_v21TurboDPMSDE.safetensors")
# Get the path to the parent directory, which should be the 'bin' directory
if platform.system()=="Windows":
bin_dir = Path(sys.executable).parent.parent/"miniconda3/condabin"
else:
bin_dir = Path(sys.executable).parent.parent/"miniconda3/bin"
if bin_dir.exists():
conda_dir = str(bin_dir/ "conda")
# For Windows, the activate script has a '.bat' extension
if os.name == 'nt':
conda_dir += '.bat'
result = subprocess.run([conda_dir, "create","--name","autosd","-y","python==3.10"])
else:
import conda.cli
conda.cli.main("create","--name","autosd","-y","python==3.10")
ASCIIColors.green("Stable diffusion installed successfully")
@ -266,6 +281,17 @@ class LollmsSD:
if platform.system() == "Windows":
ASCIIColors.info("Running on windows")
script_path = self.sd_folder / "lollms_sd.bat"
# # Get the path to the parent directory, which should be the 'bin' directory
# bin_dir = Path(python_path).parent.parent/"miniconda3/envs/autosd"
# if bin_dir.exists():
# python_path = Path(sys.executable).parent.parent/"miniconda3/envs/autosd/python"
# command = f"{python_path}"
# ASCIIColors.cyan(command)
# process = subprocess.Popen(command, stdout=subprocess.PIPE, shell=True)
# else:
# command = f'conda activate autosd && python -m xtts_api_server -o {self.output_folder} -sf {self.voice_samples_path} -p {self.xtts_base_url.split(':')[-1].replace('/','')}'
# process = subprocess.Popen(command, stdout=subprocess.PIPE, shell=True)
if share:
subprocess.Popen(str(script_path) +" --share", cwd=self.sd_folder)
else:

2
lollms/services/xtts/lollms_xtts.py
View File

@ -70,7 +70,7 @@ def install_xtts(lollms_app:LollmsApplication):
result = subprocess.run([python_path, "-m", "pip", "install", "torch==2.1.1+cu118","torchaudio==2.1.1+cu118","--index-url https://download.pytorch.org/whl/cu118"])
else:
import conda.cli
conda.cli.main("create","--name","xtts","-y")
conda.cli.main("create","--name","xtts","-y","python==3.11")
result = subprocess.run(["conda","activate","xtts", "&&", "python", "-m", "pip", "install", "--upgrade", "xtts-api-server"])