From b9ed277055a961dc4609ee0ccc05786032c2d063 Mon Sep 17 00:00:00 2001 From: DGX Date: Thu, 1 Feb 2024 13:26:12 +0000 Subject: [PATCH] pretrained path --- examples/multi_trans.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/examples/multi_trans.py b/examples/multi_trans.py index 4dd7628..ff1259b 100644 --- a/examples/multi_trans.py +++ b/examples/multi_trans.py @@ -9,7 +9,10 @@ torch.backends.cudnn.benchmark = False warnings.filterwarnings('ignore') # %% First let us spawn a stable diffusion holder. Uncomment your version of choice. -pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16") +pretrained_model_name_or_path = "stabilityai/stable-diffusion-xl-base-1.0" +# pretrained_model_name_or_path = "stabilityai/sdxl-turbo" + +pipe = AutoPipelineForText2Image.from_pretrained(pretrained_model_name_or_path, torch_dtype=torch.float16, variant="fp16") pipe.to('cuda') be = BlendingEngine(pipe)