mirror of
https://github.com/lllyasviel/Fooocus.git
synced 2026-08-16 13:13:16 +02:00
speed up lcm again
This commit is contained in:
@@ -132,6 +132,25 @@ def clip_encode_single(clip, text, verbose=False):
|
||||
return result
|
||||
|
||||
|
||||
@torch.no_grad()
|
||||
@torch.inference_mode()
|
||||
def clone_cond(conds):
|
||||
results = []
|
||||
|
||||
for c, p in conds:
|
||||
p = p["pooled_output"]
|
||||
|
||||
if isinstance(c, torch.Tensor):
|
||||
c = c.clone()
|
||||
|
||||
if isinstance(p, torch.Tensor):
|
||||
p = p.clone()
|
||||
|
||||
results.append([c, {"pooled_output": p}])
|
||||
|
||||
return results
|
||||
|
||||
|
||||
@torch.no_grad()
|
||||
@torch.inference_mode()
|
||||
def clip_encode(texts, pool_top_k=1):
|
||||
|
||||
Reference in New Issue
Block a user