Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: add new argument to overridden method to match new signature upstream

Browse Source
This commit is contained in:
Kevin Turner 2023-03-09 17:34:41 -08:00
parent 081397737b
commit faa2558e2f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/stable_diffusion/diffusers_pipeline.py
View File

@ -440,7 +440,7 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline):
"""
self._model_group.ready()
def to(self, torch_device: Optional[Union[str, torch.device]] = None):
def to(self, torch_device: Optional[Union[str, torch.device]] = None, silence_dtype_warnings=False):
# overridden method; types match the superclass.
if torch_device is None:
return self