mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
final fixups to memory_cache
- fixed backwards calculation of minimum available memory - only execute m.padding adjustment code once upon load
This commit is contained in:
@ -679,11 +679,6 @@ class Generate:
|
||||
)
|
||||
|
||||
self._set_sampler()
|
||||
|
||||
for m in self.model.modules():
|
||||
if isinstance(m, (nn.Conv2d, nn.ConvTranspose2d)):
|
||||
m._orig_padding_mode = m.padding_mode
|
||||
|
||||
self.model_name = model_name
|
||||
return self.model
|
||||
|
||||
|
Reference in New Issue
Block a user