From 12cfcddcadb2f6f698e26341798ac780ada1257b Mon Sep 17 00:00:00 2001 From: Beyondo <58893646+Beyondo@users.noreply.github.com> Date: Thu, 22 Dec 2022 14:39:22 +0300 Subject: [PATCH] Update PerformancePipeline.py --- PerformancePipeline.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/PerformancePipeline.py b/PerformancePipeline.py index d3b9489..198f53b 100644 --- a/PerformancePipeline.py +++ b/PerformancePipeline.py @@ -1,8 +1,7 @@ # PerformancePipeline Or PP for convenience (Hahah- ok, I'm sorry) from diffusers import StableDiffusionPipeline import torch, os, importlib -from hax import safety_patcher, clip_pipeline, clip_limit -importlib.reload(safety_patcher) +from hax import clip_pipeline, clip_limit importlib.reload(clip_pipeline) def from_pretrained(model_name): os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb=512"