Skip to content

GitLab

Commit be9c5e25 authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix issue with not freeing enough memory when sampling.

parent ac0758a1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
comfy/sample.py
View file @ be9c5e25
...@@ -78,7 +78,7 @@ def sample(model, noise, steps, cfg, sampler_name, scheduler, positive, negative ...@@ -78,7 +78,7 @@ def sample(model, noise, steps, cfg, sampler_name, scheduler, positive, negative
real_model = None real_model = None
models = get_additional_models(positive, negative) models = get_additional_models(positive, negative)
comfy.model_management.load_models_gpu([model] + models, comfy.model_management.batch_area_memory(noise.shape[2] * noise.shape[3])) comfy.model_management.load_models_gpu([model] + models, comfy.model_management.batch_area_memory(noise.shape[0] * noise.shape[2] * noise.shape[3]))
real_model = model.model real_model = model.model
noise = noise.to(device) noise = noise.to(device)
......
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