soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

Update diffusers_patcher.py

Browse Source
This commit is contained in:
lllyasviel 2024-01-28 10:13:57 -08:00
parent 5df7a4d5d1
commit a009e5ecae
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules_forge/diffusers_patcher.py
View File

@ -19,8 +19,9 @@ class DiffusersPatcher:
with use_patched_ops(ops.manual_cast):
with modeling_utils.no_init_weights():
self.pipeline = pipeline_class.from_pretrained(*args, **kwargs).to(dtype)
self.pipeline = pipeline_class.from_pretrained(*args, **kwargs)
self.pipeline = self.pipeline.to(device=offload_device, dtype=dtype)
self.pipeline.eval()
self.patcher = ModelPatcher(