Spaces:
Runtime error
Runtime error
Damian Stewart
commited on
Commit
·
b58675c
1
Parent(s):
ab11bdd
typo
Browse files
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 |
-
|
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,
|