diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 6071720d..b08e56c2 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -545,6 +545,30 @@ jobs: - name: Pack artifacts if: ${{ ( github.event_name == 'push' && github.ref == 'refs/heads/master' ) || github.event.inputs.create_release == 'true' }} run: | + $ErrorActionPreference = "Stop" + $dst = "build\bin" + $rocmBin = Join-Path "${env:HIP_PATH}" "bin" + $requiredRocmPaths = @( + (Join-Path $rocmBin "rocblas.dll"), + (Join-Path $rocmBin "rocblas\library") + ) + foreach ($path in $requiredRocmPaths) { + if (!(Test-Path $path)) { + throw "Missing ROCm runtime dependency: $path" + } + } + + foreach ($pattern in @("rocblas*.dll", "hipblas*.dll", "libhipblas*.dll")) { + Copy-Item -Path (Join-Path $rocmBin $pattern) -Destination $dst -Force -ErrorAction SilentlyContinue + } + + foreach ($dir in @("rocblas", "hipblaslt")) { + $src = Join-Path $rocmBin $dir + if (Test-Path $src) { + Copy-Item -Path $src -Destination $dst -Recurse -Force + } + } + 7z a sd-${{ env.BRANCH_NAME }}-${{ steps.commit.outputs.short }}-bin-win-rocm-${{ env.ROCM_VERSION }}-x64.zip .\build\bin\* - name: Upload artifacts diff --git a/src/stable-diffusion.cpp b/src/stable-diffusion.cpp index 8f4f159a..4db5a235 100644 --- a/src/stable-diffusion.cpp +++ b/src/stable-diffusion.cpp @@ -1113,7 +1113,6 @@ public: pred_type = EPS_PRED; } } else if (sd_version_is_sd3(version) || - sd_version_is_ltxav(version) || sd_version_is_wan(version) || sd_version_is_qwen_image(version) || version == VERSION_HIDREAM_O1 || @@ -1121,16 +1120,17 @@ public: sd_version_is_ernie_image(version) || sd_version_is_z_image(version)) { pred_type = FLOW_PRED; - if (sd_version_is_ltxav(version)) { - default_flow_shift = 2.37f; - } else if (sd_version_is_wan(version)) { + if (sd_version_is_wan(version)) { default_flow_shift = 5.f; } else if (sd_version_is_ernie_image(version)) { default_flow_shift = 4.f; } else { default_flow_shift = 3.f; } - } else if (sd_version_is_flux(version) || sd_version_is_longcat(version) || sd_version_is_lens(version)) { + } else if (sd_version_is_flux(version) || + sd_version_is_longcat(version) || + sd_version_is_lens(version) || + sd_version_is_ltxav(version)) { pred_type = FLUX_FLOW_PRED; default_flow_shift = 1.0f; // TODO: validate @@ -1144,6 +1144,8 @@ public: default_flow_shift = 3.0f; } else if (sd_version_is_lens(version)) { default_flow_shift = 1.83f; + } else if (sd_version_is_ltxav(version)) { + default_flow_shift = 2.37f; } } else if (sd_version_is_flux2(version)) { pred_type = FLUX2_FLOW_PRED;