edm2_model=None for gora batch

67372a · 67372a · commit 5c053498133d · 2026-05-15T15:00:55.000-05:00
diff --git a/train_network.py b/train_network.py
@@ -1415,6 +1415,7 @@ def gora_forward_fn(batch):
                     is_train=True,
                     train_text_encoder=train_text_encoder,
                     train_unet=train_unet,
+                    edm2_model=None,
                 )
 
             network.prepare_gora(

Original file line number	Diff line number	Diff line change
`@@ -1415,6 +1415,7 @@ def gora_forward_fn(batch):`
`1415`	`1415`	`is_train=True,`
`1416`	`1416`	`train_text_encoder=train_text_encoder,`
`1417`	`1417`	`train_unet=train_unet,`
	`1418`	`+ edm2_model=None,`
`1418`	`1419`	`)`
`1419`	`1420`
`1420`	`1421`	`network.prepare_gora(`