Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
4146ea9
1
Parent(s):
689fe99
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
|
| 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 # 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)
|