diff --git a/ui/easydiffusion/app.py b/ui/easydiffusion/app.py index e2c190f8..91608dfb 100644 --- a/ui/easydiffusion/app.py +++ b/ui/easydiffusion/app.py @@ -172,6 +172,7 @@ getConfig.__test_diffusers_on_startup = None def setConfig(config): try: # config.yaml config_yaml_path = os.path.join(CONFIG_DIR, "..", "config.yaml") + config_yaml_path = os.path.abspath(config_yaml_path) yaml = YAML() if not hasattr(config, "_yaml_comment"): diff --git a/ui/easydiffusion/package_manager.py b/ui/easydiffusion/package_manager.py index 72479379..c28a58a1 100644 --- a/ui/easydiffusion/package_manager.py +++ b/ui/easydiffusion/package_manager.py @@ -12,9 +12,9 @@ from easydiffusion import app manifest = { "tensorrt": { "install": [ - "nvidia-cudnn --pre --extra-index-url=https://pypi.nvidia.com --trusted-host pypi.nvidia.com", - "tensorrt-libs --pre --extra-index-url=https://pypi.nvidia.com --trusted-host pypi.nvidia.com", - "tensorrt --pre --extra-index-url=https://pypi.nvidia.com --trusted-host pypi.nvidia.com", + "wheel", + "nvidia-cudnn-cu11==8.9.4.25", + "tensorrt==9.0.0.post11.dev1 --pre --extra-index-url=https://pypi.nvidia.com --trusted-host pypi.nvidia.com", ], "uninstall": ["tensorrt"], # TODO also uninstall tensorrt-libs and nvidia-cudnn, but do it upon restarting (avoid 'file in use' error)