Revert "feat: add clip skip handling (#2999)" (#3008)

This reverts commit cc58fe5270.
This commit is contained in:
Manuel Schmid
2024-05-26 22:07:44 +02:00
committed by GitHub
parent de34023c79
commit 989a1ad52b
6 changed files with 12 additions and 41 deletions
-11
View File
@@ -201,17 +201,6 @@ def clip_encode(texts, pool_top_k=1):
return [[torch.cat(cond_list, dim=1), {"pooled_output": pooled_acc}]]
@torch.no_grad()
@torch.inference_mode()
def set_clip_skip(clip_skip: int):
global final_clip
if final_clip is None:
return
final_clip.clip_layer(-abs(clip_skip))
return
@torch.no_grad()
@torch.inference_mode()
def clear_all_caches():