This commit is contained in:
2023-02-18 21:42:38 +08:00
parent 26472323f9
commit d64260d5b0

View File

@@ -345,8 +345,10 @@ def main_dev(opt):
if isinstance(prompts, tuple):
prompts = list(prompts)
with torch.no_grad(), additional_context:
for _ in range(3):
c = model.get_learned_conditioning(prompts)
#for _ in range(3):
# c = model.get_learned_conditioning(prompts)
c = model.get_learned_conditioning(prompts)
print(c)
samples_ddim, _ = sampler.sample(S=5,
conditioning=c,
batch_size=batch_size,