mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-04-26 02:12:31 +08:00
Merge upstream/master, keep local README.md
This commit is contained in:
commit
7d397b7cb7
20
.github/workflows/test-ci.yml
vendored
20
.github/workflows/test-ci.yml
vendored
@ -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: ""
|
||||||
|
|||||||
@ -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)
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user