lower ram usage

2026-04-08 22:35:09 +00:00 · 2023-02-08 18:31:27 +08:00
parent d591891048
commit 938bd71844
1 changed files with 3 additions and 1 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -150,7 +150,9 @@ def train(args):

  # モデルを読み込む
  text_encoder, vae, unet, _ = train_util.load_target_model(args, weight_dtype)
-
+  # unnecessary, but work on low-ram device
+  text_encoder.to("cuda")
+  unet.to("cuda")
  # モデルに xformers とか memory efficient attention を組み込む
  train_util.replace_unet_modules(unet, args.mem_eff_attn, args.xformers)