kopia lustrzana https://github.com/lzzcd001/MeshDiffusion
Update sampling.py
rodzic
f77e148614
commit
12c251df37
|
@ -390,7 +390,7 @@ def get_pc_sampler(sde, shape, predictor, corrector, inverse_scaler, snr,
|
||||||
n_steps=n_steps)
|
n_steps=n_steps)
|
||||||
|
|
||||||
def pc_sampler(model,
|
def pc_sampler(model,
|
||||||
partial=None, partial_grid_mask=None, partial_channel=0,
|
partial=None, partial_mask=None, partial_channel=0,
|
||||||
freeze_iters=None):
|
freeze_iters=None):
|
||||||
""" The PC sampler funciton.
|
""" The PC sampler funciton.
|
||||||
|
|
||||||
|
@ -520,7 +520,7 @@ def get_ddim_sampler(sde, shape, predictor, inverse_scaler, n_steps=1,
|
||||||
continuous=False)
|
continuous=False)
|
||||||
|
|
||||||
def ddim_sampler(model, schedule='quad', num_steps=100, x0=None,
|
def ddim_sampler(model, schedule='quad', num_steps=100, x0=None,
|
||||||
partial=None, partial_grid_mask=None, partial_channel=0):
|
partial=None, partial_mask=None, partial_channel=0):
|
||||||
""" The PC sampler funciton.
|
""" The PC sampler funciton.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
|
|
Ładowanie…
Reference in New Issue