kevinjdolan
commented
2 years ago I was able to disable the progress produced by the guided-diffusion library through a monkey-patched hack:
def _disable_progress():
from guided_diffusion.gaussian_diffusion import GaussianDiffusion
orig_ddim = GaussianDiffusion.ddim_sample_loop_progressive
orig_plms = GaussianDiffusion.plms_sample_loop_progressive
def new_ddim(*args, **kwargs):
kwargs['progress'] = False
return orig_ddim(*args, **kwargs)
def new_plms(*args, **kwargs):
kwargs['progress'] = False
return orig_plms(*args, **kwargs)
GaussianDiffusion.ddim_sample_loop_progressive = new_ddim
GaussianDiffusion.plms_sample_loop_progressive = new_plmsThis succeeds at removing the progress bars.
A cleaner way would be to accept an environment variable, perhaps here:
runner.py#373
show_progress = not os.environ.get('DISCOART_DISABLE_PROGRESS')
if args.diffusion_sampling_mode == 'ddim':
samples = sample_fn(
model,
(args.batch_size, 3, side_y, side_x),
clip_denoised=args.clip_denoised,
model_kwargs={},
cond_fn=cond_fn,
progress=show_progress,
skip_timesteps=skip_steps,
init_image=init,
randomize_class=args.randomize_class,
eta=args.eta,
transformation_fn=lambda x: symmetry_transformation_fn(
x, args.use_horizontal_symmetry, args.use_vertical_symmetry
),
transformation_percent=args.transformation_percent,
)
else:
samples = sample_fn(
model,
(args.batch_size, 3, side_y, side_x),
clip_denoised=args.clip_denoised,
model_kwargs={},
cond_fn=cond_fn,
progress=show_progress,
skip_timesteps=skip_steps,
init_image=init,
randomize_class=args.randomize_class,
order=2,
)Apparently this does not solve my problem, however! I am seeing a lot of newlines being generated but I am not sure where they are coming from, and this is not an issue when I am running it interactively. Odd behavior!
entmike
hanxiao
I was trying to search through the code to figure out where the progress bars are being generated, but I could not locate them. I assume they are being powered by TQDM.
I am getting frustrating output when the logs are output to a file in the context of a Docker container running on Kubernetes. I would like to either remove the TQDM rendering altogether, or modify so that it is not producing the input characters intended for a live console.
Is there a documented way to control the TQDM rendering? If not, can you point me in the right direction for how to control these?