Spaces:

realfill-library
/

RealFill-Training-UI

Runtime error

thuanz123 commited on Oct 22, 2023

Commit

095fa61

1 Parent(s): 6691c9c

Update train_realfill.py

Files changed (1) hide show

train_realfill.py CHANGED Viewed

@@ -108,7 +108,6 @@ def log_validation(
         args.pretrained_model_name_or_path,
         tokenizer=tokenizer,
         revision=args.revision,
-        torch_dtype=weight_dtype,
     )
     # set `keep_fp32_wrapper` to True because we do not want to remove

         args.pretrained_model_name_or_path,
         tokenizer=tokenizer,
         revision=args.revision,
     )
     # set `keep_fp32_wrapper` to True because we do not want to remove