From d64260d5b033636cd74aad212c2a8d7acb9c2de0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= Date: Sat, 18 Feb 2023 21:42:38 +0800 Subject: [PATCH] prompt --- server.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/server.py b/server.py index 0355bd7..a1eda1d 100644 --- a/server.py +++ b/server.py @@ -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,