Damian Stewart commited on
Commit
b58675c
·
1 Parent(s): ab11bdd
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -20,7 +20,7 @@ def train(repo_id_or_path, img_size, prompt, modules, freeze_modules, iterations
20
  diffuser.train()
21
  finetuner = FineTunedModel(diffuser, modules, frozen_modules=freeze_modules)
22
  if use_adamw8bit:
23
- use print("using AdamW 8Bit optimizer")
24
  import bitsandbytes as bnb
25
  optimizer = bnb.optim.AdamW8bit(finetuner.parameters(),
26
  lr=lr,
 
20
  diffuser.train()
21
  finetuner = FineTunedModel(diffuser, modules, frozen_modules=freeze_modules)
22
  if use_adamw8bit:
23
+ print("using AdamW 8Bit optimizer")
24
  import bitsandbytes as bnb
25
  optimizer = bnb.optim.AdamW8bit(finetuner.parameters(),
26
  lr=lr,