dkebudi commited on
Commit
e8383f5
·
verified ·
1 Parent(s): 439be08

pipeline args all updated to have hard coded dimensions in validation and prediction

Browse files
train_dreambooth_lora_sdxl_advanced.py CHANGED
@@ -2346,7 +2346,7 @@ def main(args):
2346
  variant=args.variant,
2347
  torch_dtype=weight_dtype,
2348
  )
2349
- pipeline_args = {"prompt": args.validation_prompt}
2350
 
2351
  images = log_validation(
2352
  pipeline,
 
2346
  variant=args.variant,
2347
  torch_dtype=weight_dtype,
2348
  )
2349
+ pipeline_args = {"prompt": args.validation_prompt, "width":1920, "height":1080}
2350
 
2351
  images = log_validation(
2352
  pipeline,