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

diffusers(ModelCache): stopgap to make from_cpu compatible with diffusers

Browse Source
This commit is contained in:
Kevin Turner 2022-12-09 10:23:39 -08:00
parent 9199d698f8
commit c28f56daec
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ldm/invoke/model_cache.py
View File

@ -464,9 +464,12 @@ class ModelCache(object):
def _model_from_cpu(self,model): def _model_from_cpu(self,model):
if self.device != 'cpu': if self.device != 'cpu':
model.to(self.device) model.to(self.device)
try:
model.first_stage_model.to(self.device) model.first_stage_model.to(self.device)
model.cond_stage_model.to(self.device) model.cond_stage_model.to(self.device)
model.cond_stage_model.device = self.device model.cond_stage_model.device = self.device
except AttributeError as e:
warnings.warn(f"TODO: clean up legacy model-management: {e}")
return model return model
def _pop_oldest_model(self): def _pop_oldest_model(self):