diff options
| author | Volpeon <git@volpeon.ink> | 2023-06-22 07:33:29 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-06-22 07:33:29 +0200 |
| commit | 186a69104530610f8c2b924f79a04f941e5238c8 (patch) | |
| tree | f04de211c4f33151c5163be222f7297087edb7d4 /train_ti.py | |
| parent | Update (diff) | |
| download | textual-inversion-diff-186a69104530610f8c2b924f79a04f941e5238c8.tar.gz textual-inversion-diff-186a69104530610f8c2b924f79a04f941e5238c8.tar.bz2 textual-inversion-diff-186a69104530610f8c2b924f79a04f941e5238c8.zip | |
Remove convnext
Diffstat (limited to 'train_ti.py')
| -rw-r--r-- | train_ti.py | 10 |
1 files changed, 0 insertions, 10 deletions
diff --git a/train_ti.py b/train_ti.py index da0c03e..7d1ef19 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -698,16 +698,6 @@ def main(): | |||
| 698 | unet.enable_gradient_checkpointing() | 698 | unet.enable_gradient_checkpointing() |
| 699 | text_encoder.gradient_checkpointing_enable() | 699 | text_encoder.gradient_checkpointing_enable() |
| 700 | 700 | ||
| 701 | # convnext = create_model( | ||
| 702 | # "convnext_tiny", | ||
| 703 | # pretrained=False, | ||
| 704 | # num_classes=3, | ||
| 705 | # drop_path_rate=0.0, | ||
| 706 | # ) | ||
| 707 | # convnext.to(accelerator.device, dtype=weight_dtype) | ||
| 708 | # convnext.requires_grad_(False) | ||
| 709 | # convnext.eval() | ||
| 710 | |||
| 711 | if len(args.alias_tokens) != 0: | 701 | if len(args.alias_tokens) != 0: |
| 712 | alias_placeholder_tokens = args.alias_tokens[::2] | 702 | alias_placeholder_tokens = args.alias_tokens[::2] |
| 713 | alias_initializer_tokens = args.alias_tokens[1::2] | 703 | alias_initializer_tokens = args.alias_tokens[1::2] |