extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-20 01:48:21 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1538 from easydiffusion/beta

Browse Source 
Beta
This commit is contained in:
cmdr2 2023-08-30 15:35:21 +05:30 committed by GitHub
commit 1a6e0234b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ui/easydiffusion/app.py
View File

@ -172,6 +172,7 @@ getConfig.__test_diffusers_on_startup = None
def setConfig(config): def setConfig(config):
try: # config.yaml try: # config.yaml
config_yaml_path = os.path.join(CONFIG_DIR, "..", "config.yaml") config_yaml_path = os.path.join(CONFIG_DIR, "..", "config.yaml")
config_yaml_path = os.path.abspath(config_yaml_path)
yaml = YAML() yaml = YAML()
if not hasattr(config, "_yaml_comment"): if not hasattr(config, "_yaml_comment"):

6
ui/easydiffusion/package_manager.py
View File

@ -12,9 +12,9 @@ from easydiffusion import app
manifest = { manifest = {
"tensorrt": { "tensorrt": {
"install": [ "install": [
"nvidia-cudnn --pre --extra-index-url=https://pypi.nvidia.com --trusted-host pypi.nvidia.com", "wheel",
"tensorrt-libs --pre --extra-index-url=https://pypi.nvidia.com --trusted-host pypi.nvidia.com", "nvidia-cudnn-cu11==8.9.4.25",
"tensorrt --pre --extra-index-url=https://pypi.nvidia.com --trusted-host pypi.nvidia.com", "tensorrt==9.0.0.post11.dev1 --pre --extra-index-url=https://pypi.nvidia.com --trusted-host pypi.nvidia.com",
], ],
"uninstall": ["tensorrt"], "uninstall": ["tensorrt"],
# TODO also uninstall tensorrt-libs and nvidia-cudnn, but do it upon restarting (avoid 'file in use' error) # TODO also uninstall tensorrt-libs and nvidia-cudnn, but do it upon restarting (avoid 'file in use' error)