This commit is contained in:
Patrick von Platen
2023-07-25 21:50:35 +00:00
parent a9d239ab51
commit e6baac124d
-4
View File
@@ -994,10 +994,6 @@ def apply_token_merging(sd_model, token_merging_ratio):
if current_token_merging_ratio > 0:
tomesd.remove_patch(sd_model)
if sd_model.__class__ == PriorPipeline:
# token merging is not supported for PriorPipelines currently
return
if token_merging_ratio > 0:
tomesd.apply_patch(
sd_model,