Skip to content

Commit 04abab5

Browse files
committed
merge
Signed-off-by: Jingyu Xin <jingyux@nvidia.com>
2 parents 6685adf + 920fd4f commit 04abab5

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

examples/diffusers/quantization/quantize.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -488,8 +488,7 @@ def main() -> None:
488488

489489
model_type = ModelType(args.model)
490490
if args.backbone is None:
491-
args.backbone = MODEL_DEFAULTS[model_type]["backbone"]
492-
491+
args.backbone = [MODEL_DEFAULTS[model_type]["backbone"]]
493492
s = time.time()
494493

495494
model_dtype = {"default": DataType(args.model_dtype).torch_dtype}

0 commit comments

Comments
 (0)