Commit
·
01429f1
1
Parent(s):
8a99c28
add inference
Browse files
uni_pc.py
CHANGED
|
@@ -549,7 +549,7 @@ class UniPC:
|
|
| 549 |
if not self.use_afs:
|
| 550 |
fir_output = self.noise_prediction_fn(x, vec_t)
|
| 551 |
else:
|
| 552 |
-
fir_output = x * 0.
|
| 553 |
if npnet_x is not None and npnet_scale is not None:
|
| 554 |
fir_output = npnet_x
|
| 555 |
# fir_output = fir_output - npnet_scale * (npnet_out - fir_output) #guidance_scale * (noise - noise_uncond)
|
|
|
|
| 549 |
if not self.use_afs:
|
| 550 |
fir_output = self.noise_prediction_fn(x, vec_t)
|
| 551 |
else:
|
| 552 |
+
fir_output = x * 0.98 # ultilize npnet there in the future
|
| 553 |
if npnet_x is not None and npnet_scale is not None:
|
| 554 |
fir_output = npnet_x
|
| 555 |
# fir_output = fir_output - npnet_scale * (npnet_out - fir_output) #guidance_scale * (noise - noise_uncond)
|