Merge upstream/master, keep local README.md

This commit is contained in:
GitHub Actions 2025-11-11 00:36:09 +00:00
commit 7d397b7cb7
2 changed files with 13 additions and 11 deletions

View File

@ -21,14 +21,15 @@ jobs:
fail-fast: false fail-fast: false
matrix: matrix:
# os: [macos, linux, windows] # os: [macos, linux, windows]
os: [macos, linux] # os: [macos, linux]
python_version: ["3.9", "3.10", "3.11", "3.12"] os: [linux]
python_version: ["3.10", "3.11", "3.12"]
cuda_version: ["12.1"] cuda_version: ["12.1"]
torch_version: ["stable"] torch_version: ["stable"]
include: include:
- os: macos # - os: macos
runner_label: [self-hosted, macOS] # runner_label: [self-hosted, macOS]
flags: "--use-pytorch-cross-attention" # flags: "--use-pytorch-cross-attention"
- os: linux - os: linux
runner_label: [self-hosted, Linux] runner_label: [self-hosted, Linux]
flags: "" flags: ""
@ -73,14 +74,15 @@ jobs:
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
os: [macos, linux] # os: [macos, linux]
os: [linux]
python_version: ["3.11"] python_version: ["3.11"]
cuda_version: ["12.1"] cuda_version: ["12.1"]
torch_version: ["nightly"] torch_version: ["nightly"]
include: include:
- os: macos # - os: macos
runner_label: [self-hosted, macOS] # runner_label: [self-hosted, macOS]
flags: "--use-pytorch-cross-attention" # flags: "--use-pytorch-cross-attention"
- os: linux - os: linux
runner_label: [self-hosted, Linux] runner_label: [self-hosted, Linux]
flags: "" flags: ""

View File

@ -110,9 +110,9 @@ def cast_bias_weight(s, input=None, dtype=None, device=None, bias_dtype=None, of
for f in s.bias_function: for f in s.bias_function:
bias = f(bias) bias = f(bias)
weight = weight.to(dtype=dtype) if weight_has_function or weight.dtype != dtype:
if weight_has_function:
with wf_context: with wf_context:
weight = weight.to(dtype=dtype)
for f in s.weight_function: for f in s.weight_function:
weight = f(weight) weight = f(weight)