i

modules/core.py

@@ -31,13 +31,18 @@ def encode_prompt_condition(clip, prompt):
     return opCLIPTextEncode.encode(clip=clip, text=prompt)[0]
 
 
+@torch.no_grad()
+def generate_empty_latent(width=1024, height=1024, batch_size=1):
+    return opEmptyLatentImage.generate(width=width, height=height, batch_size=batch_size)[0]
+
+
 @torch.no_grad()
 def decode_vae(vae, latent_image):
     return opVAEDecode.decode(samples=latent_image, vae=vae)[0]
 
 
 @torch.no_grad()
-def ksample(model, positive_condition, negative_condition, latent_image, add_noise=True, noise_seed=None, steps=25, cfg=9,
+def ksample(unet, positive_condition, negative_condition, latent_image, add_noise=True, noise_seed=None, steps=25, cfg=9,
             sampler_name='euler_ancestral', scheduler='normal', start_at_step=None, end_at_step=None,
             return_with_leftover_noise=False):
     return opKSamplerAdvanced.sample(
@@ -50,7 +55,7 @@ def ksample(model, positive_condition, negative_condition, latent_image, add_noi
         start_at_step=0 if start_at_step is None else start_at_step,
         end_at_step=steps if end_at_step is None else end_at_step,
         return_with_leftover_noise='enable' if return_with_leftover_noise else 'disable',
-        model=model,
+        model=unet,
         positive=positive_condition,
         negative=negative_condition,
         latent_image=latent_image,