Add a node to set CLIP skip.

Use a more simple way to detect if the model is -v prediction.
This commit is contained in:
comfyanonymous
2023-03-03 13:04:36 -05:00
parent fed315a76a
commit 4215206281
3 changed files with 27 additions and 9 deletions

View File

@@ -81,7 +81,7 @@ class DDPM(torch.nn.Module):
super().__init__()
assert parameterization in ["eps", "x0", "v"], 'currently only supporting "eps" and "x0" and "v"'
self.parameterization = parameterization
# print(f"{self.__class__.__name__}: Running in {self.parameterization}-prediction mode")
print(f"{self.__class__.__name__}: Running in {self.parameterization}-prediction mode")
self.cond_stage_model = None
self.clip_denoised = clip_denoised
self.log_every_t = log_every_t