fix upscaler compile, diffusers enum, svd

This commit is contained in:
Vladimir Mandic
2024-03-09 08:35:18 -05:00
parent 867677115c
commit 6cb20edf7e
6 changed files with 85 additions and 64 deletions
+1 -3
View File
@@ -62,9 +62,7 @@ class Script(scripts.Script):
p.do_not_save_grid = True
if c == 'StableVideoDiffusionPipeline':
p.sampler_name = 'Default' # svd does not support non-default sampler
p.task_args['output_type'] = 'np'
else:
p.task_args['output_type'] = 'pil'
p.task_args['output_type'] = 'pil'
p.task_args['generator'] = torch.manual_seed(p.seed) # svd does not support gpu based generator
p.task_args['width'] = p.width
p.task_args['height'] = p.height