diff --git a/README.md b/README.md index 4f399a332c..1b02ced2c9 100644 --- a/README.md +++ b/README.md @@ -145,7 +145,7 @@ not supported. _For Linux with an AMD GPU:_ ```sh - pip install InvokeAI --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.2 + pip install InvokeAI --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 ``` _For Macintoshes, either Intel or M1/M2:_ diff --git a/docs/installation/010_INSTALL_AUTOMATED.md b/docs/installation/010_INSTALL_AUTOMATED.md index 228c0ae9a4..83b4415394 100644 --- a/docs/installation/010_INSTALL_AUTOMATED.md +++ b/docs/installation/010_INSTALL_AUTOMATED.md @@ -417,7 +417,7 @@ Then type the following commands: === "AMD System" ```bash - pip install torch torchvision --force-reinstall --extra-index-url https://download.pytorch.org/whl/rocm5.2 + pip install torch torchvision --force-reinstall --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 ``` ### Corrupted configuration file diff --git a/docs/installation/020_INSTALL_MANUAL.md b/docs/installation/020_INSTALL_MANUAL.md index 401560e76c..657e3f055d 100644 --- a/docs/installation/020_INSTALL_MANUAL.md +++ b/docs/installation/020_INSTALL_MANUAL.md @@ -154,7 +154,7 @@ manager, please follow these steps: === "ROCm (AMD)" ```bash - pip install InvokeAI --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.2 + pip install InvokeAI --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 ``` === "CPU (Intel Macs & non-GPU systems)" @@ -315,7 +315,7 @@ installation protocol (important!) === "ROCm (AMD)" ```bash - pip install -e . --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.2 + pip install -e . --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 ``` === "CPU (Intel Macs & non-GPU systems)" diff --git a/docs/installation/030_INSTALL_CUDA_AND_ROCM.md b/docs/installation/030_INSTALL_CUDA_AND_ROCM.md index 8ce690ca64..3d3445e3a0 100644 --- a/docs/installation/030_INSTALL_CUDA_AND_ROCM.md +++ b/docs/installation/030_INSTALL_CUDA_AND_ROCM.md @@ -110,7 +110,7 @@ recipes are available When installing torch and torchvision manually with `pip`, remember to provide the argument `--extra-index-url -https://download.pytorch.org/whl/rocm5.2` as described in the [Manual +https://download.pytorch.org/whl/rocm5.4.2` as described in the [Manual Installation Guide](020_INSTALL_MANUAL.md). This will be done automatically for you if you use the installer diff --git a/invokeai/backend/model_management/convert_ckpt_to_diffusers.py b/invokeai/backend/model_management/convert_ckpt_to_diffusers.py index 3547fcee7b..b46586611d 100644 --- a/invokeai/backend/model_management/convert_ckpt_to_diffusers.py +++ b/invokeai/backend/model_management/convert_ckpt_to_diffusers.py @@ -1264,10 +1264,10 @@ def load_pipeline_from_original_stable_diffusion_ckpt( cache_dir=cache_dir, ) pipe = pipeline_class( - vae=vae, - text_encoder=text_model, + vae=vae.to(precision), + text_encoder=text_model.to(precision), tokenizer=tokenizer, - unet=unet, + unet=unet.to(precision), scheduler=scheduler, safety_checker=None, feature_extractor=None, diff --git a/pyproject.toml b/pyproject.toml index 393db0cae4..1da7c7bd77 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -45,7 +45,7 @@ dependencies = [ "einops", "eventlet", "facexlib", - "fastapi==0.94.1", + "fastapi==0.88.0", "fastapi-events==0.8.0", "fastapi-socketio==0.0.10", "flask==2.1.3", @@ -160,4 +160,3 @@ output = "coverage/index.xml" [flake8] max-line-length = 120 -