blepping
7ced6b1626
Remove sigma_offset YAML parameter Add sigma_dishonesty_factor[_guidance] YAML parameters Internal cleanups and refactoring Minor documentation improvements
25 lines
637 B
Python
25 lines
637 B
Python
from __future__ import annotations
|
|
|
|
import math
|
|
|
|
from comfy import model_management
|
|
|
|
|
|
def ensure_model(model):
|
|
mp = model.inner_model.model_patcher
|
|
if model_management.LoadedModel(mp) in model_management.current_loaded_models:
|
|
return
|
|
model_management.load_models_gpu((mp,))
|
|
|
|
|
|
def fallback(val, default, *, exclude=None, default_is_fun=False):
|
|
return val if val is not exclude else (default() if default_is_fun else default)
|
|
|
|
|
|
def scale_dim(n, factor=1.0, *, increment=64) -> int:
|
|
return math.ceil((n * factor) / increment) * increment
|
|
|
|
|
|
def sigma_to_float(sigma):
|
|
return sigma.detach().cpu().max().item()
|