[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
a4694d1010
commit
14eba1c887
1 changed files with 6 additions and 1 deletions
|
|
@ -414,7 +414,12 @@ def _encode_once(bag) -> None:
|
|||
|
||||
|
||||
# ── VAE loading + latent shape (reused from quant_accuracy_sweep) ──────────────
|
||||
def _load_vae(repo: str, device: str, *, force_fp32: bool = False):
|
||||
def _load_vae(
|
||||
repo: str,
|
||||
device: str,
|
||||
*,
|
||||
force_fp32: bool = False,
|
||||
):
|
||||
import torch
|
||||
|
||||
diffusers = _import_diffusers()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue