gameui/drawing
2
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

prompt

Browse Source
This commit is contained in:
桜華
2023-02-18 21:42:38 +08:00
parent 26472323f9
commit d64260d5b0
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
server.py
View File

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