18 lines
514 B
Python
18 lines
514 B
Python
import comfy.model_management
|
|
|
|
|
|
def get_base_model(model):
|
|
base_model = model
|
|
while hasattr(base_model, 'inner_model'):
|
|
base_model = base_model.inner_model
|
|
return base_model
|
|
|
|
|
|
def prepare_ref_latents(model, ref_latent):
|
|
base_model = model.model # get_base_model(model)
|
|
ref_latent = ref_latent['samples'].clone()
|
|
ref_latent = base_model.process_latent_in(ref_latent)
|
|
device = comfy.model_management.get_torch_device()
|
|
ref_latent = ref_latent.to(device)
|
|
return ref_latent
|