storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-13 23:53:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

address review

Browse Source
This commit is contained in:
Aryan
2024-12-23 15:11:03 +01:00
parent 1c8f33967c
commit 2858346b27
2 changed files with 0 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/args.py
-4
View File
@@ -26,7 +26,6 @@ class Args:
text_encoder_2_dtype: torch.dtype = torch.bfloat16
text_encoder_3_dtype: torch.dtype = torch.bfloat16
transformer_dtype: torch.dtype = torch.bfloat16
unet_dtype: torch.dtype = torch.bfloat16
vae_dtype: torch.dtype = torch.bfloat16
# Dataset arguments
@@ -126,7 +125,6 @@ class Args:
"text_encoder_2_dtype": self.text_encoder_2_dtype,
"text_encoder_3_dtype": self.text_encoder_3_dtype,
"transformer_dtype": self.transformer_dtype,
"unet_dtype": self.unet_dtype,
"vae_dtype": self.vae_dtype,
},
"dataset_arguments": {
@@ -255,7 +253,6 @@ def _add_model_arguments(parser: argparse.ArgumentParser) -> None:
parser.add_argument("--text_encoder_2_dtype", type=str, default="bf16", help="Data type for the text encoder 2.")
parser.add_argument("--text_encoder_3_dtype", type=str, default="bf16", help="Data type for the text encoder 3.")
parser.add_argument("--transformer_dtype", type=str, default="bf16", help="Data type for the transformer model.")
parser.add_argument("--unet_dtype", type=str, default="bf16", help="Data type for the U-Net model.")
parser.add_argument("--vae_dtype", type=str, default="bf16", help="Data type for the VAE model.")
@@ -693,7 +690,6 @@ def _map_to_args_type(args: Dict[str, Any]) -> Args:
result_args.text_encoder_2_dtype = _DTYPE_MAP[args.text_encoder_2_dtype]
result_args.text_encoder_3_dtype = _DTYPE_MAP[args.text_encoder_3_dtype]
result_args.transformer_dtype = _DTYPE_MAP[args.transformer_dtype]
result_args.unet_dtype = _DTYPE_MAP[args.unet_dtype]
result_args.vae_dtype = _DTYPE_MAP[args.vae_dtype]
# Dataset arguments
finetrainers/trainer.py
-1
View File
@@ -117,7 +117,6 @@ class Trainer:
"text_encoder_2_dtype": self.args.text_encoder_2_dtype,
"text_encoder_3_dtype": self.args.text_encoder_3_dtype,
"transformer_dtype": self.args.transformer_dtype,
"unet_dtype": self.args.unet_dtype,
"vae_dtype": self.args.vae_dtype,
"revision": self.args.revision,
"cache_dir": self.args.cache_dir,