This commit is contained in:
Ftps 2023-04-02 19:23:03 +09:00
parent 3b2e28a137
commit d6933311e7

View file

@ -311,7 +311,7 @@ def main(opt):
prompts = data[0] prompts = data[0]
print("Running a forward pass to initialize optimizations") print("Running a forward pass to initialize optimizations")
uc = Nonede uc = None
if opt.scale != 1.0: if opt.scale != 1.0:
uc = model.get_learned_conditioning(batch_size * [""]) uc = model.get_learned_conditioning(batch_size * [""])
if isinstance(prompts, tuple): if isinstance(prompts, tuple):