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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user