Spaces:

baulab
/

ConceptSliders

Runtime error

RohitGandikota commited on Nov 28, 2023

Commit

94e1b95

1 Parent(s): 9a7d487

fixing training

Files changed (1) hide show

trainscripts/textsliders/demotrain.py CHANGED Viewed

@@ -426,7 +426,7 @@ def train_xl(target, positive, negative, lr, iterations, config_file, rank, devi
     else:
         attributes = []
     config.network.alpha = 1.0
-    config.network.rank = rank
     config.save.path += f'/{config.save.name}'

     else:
         attributes = []
     config.network.alpha = 1.0
+    config.network.rank = int(rank)
     config.save.path += f'/{config.save.name}'