fix issues with hypernetwork training

AIRI-AUBE · Feb 17, 2023 · 633095a · 633095a
1 parent d4c32f0
commit 633095a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/modules/ui.py b/modules/ui.py
@@ -1980,6 +1980,7 @@ def sagemaker_train_embedding(
 
                 def sagemaker_train_hypernetwork(
                         username,
+                        sd_model_checkpoint,
                         new_hypernetwork_name,
                         new_hypernetwork_sizes,
                         new_hypernetwork_layer_structure,