From 954a6b260fa9d414c8abc0774cf51baa9546bf01 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 4 Dec 2023 11:02:46 +0000 Subject: [PATCH] [From single file] remove depr warning --- src/diffusers/loaders/single_file.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/loaders/single_file.py b/src/diffusers/loaders/single_file.py index 8c63c4cf59a5..bf100e7f2c81 100644 --- a/src/diffusers/loaders/single_file.py +++ b/src/diffusers/loaders/single_file.py @@ -282,7 +282,7 @@ def from_single_file(cls, pretrained_model_link_or_path, **kwargs): ) if torch_dtype is not None: - pipe.to(torch_dtype=torch_dtype) + pipe.to(dtype=torch_dtype) return pipe