From f10b8948c3c5b1c95ce47d7b80908438a912e0e9 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Sat, 11 Feb 2023 04:34:58 -0500
Subject: [PATCH] 768-v support for uni_pc sampler.

---
 comfy/extra_samplers/uni_pc.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/comfy/extra_samplers/uni_pc.py b/comfy/extra_samplers/uni_pc.py
index 7373cba1..cc629912 100644
--- a/comfy/extra_samplers/uni_pc.py
+++ b/comfy/extra_samplers/uni_pc.py
@@ -835,12 +835,16 @@ def sample_unipc(model, noise, image, sigmas, sampling_function, extra_args=None
 
         device = noise.device
 
+        if model.inner_model.parameterization == "v":
+            model_type = "v"
+        else:
+            model_type = "noise"
 
         model_fn = model_wrapper(
             model.inner_model.apply_model,
             sampling_function,
             ns,
-            model_type="noise",
+            model_type=model_type,
             guidance_type="uncond",
             model_kwargs=extra_args,
         )