Skip to content

GitLab

"docs/source/vscode:/vscode.git/clone" did not exist on "ea1b4ea7ca5fa7ab3d2094a70a35481f2a1036c2"
Commit ca9d300a authored by comfyanonymous's avatar comfyanonymous
Browse files

Better estimation for memory usage during audio VAE encoding/decoding.

parent 746a0410
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
comfy/sd.py
View file @ ca9d300a
...@@ -237,8 +237,8 @@ class VAE: ...@@ -237,8 +237,8 @@ class VAE:
decoder_config={'target': "comfy.ldm.modules.diffusionmodules.model.Decoder", 'params': ddconfig}) decoder_config={'target': "comfy.ldm.modules.diffusionmodules.model.Decoder", 'params': ddconfig})
elif "decoder.layers.0.weight_v" in sd: elif "decoder.layers.0.weight_v" in sd:
self.first_stage_model = AudioOobleckVAE() self.first_stage_model = AudioOobleckVAE()
self.memory_used_encode = lambda shape, dtype: (1767 * shape[2]) * model_management.dtype_size(dtype) #TODO: tweak for the audio VAE self.memory_used_encode = lambda shape, dtype: (1000 * shape[2]) * model_management.dtype_size(dtype)
self.memory_used_decode = lambda shape, dtype: (2178 * shape[2] * 64) * model_management.dtype_size(dtype) self.memory_used_decode = lambda shape, dtype: (1000 * shape[2] * 2048) * model_management.dtype_size(dtype)
self.latent_channels = 64 self.latent_channels = 64
self.output_channels = 2 self.output_channels = 2
self.upscale_ratio = 2048 self.upscale_ratio = 2048
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment