Fixed issues with adapter and disabled telementry for now

This commit is contained in:
Jaret Burkett
2023-09-24 12:51:29 -06:00
parent abf7cd221d
commit 76c764af49
2 changed files with 6 additions and 1 deletions

3
run.py
View File

@@ -6,6 +6,9 @@ sys.path.insert(0, os.getcwd())
# must come before ANY torch or fastai imports
# import toolkit.cuda_malloc
# turn off diffusers telemetry until I can figure out how to make it opt-in
os.environ['DISABLE_TELEMETRY'] = 'YES'
# check if we have DEBUG_TOOLKIT in env
if os.environ.get("DEBUG_TOOLKIT", "0") == "1":
# set torch to trace mode

View File

@@ -310,6 +310,9 @@ class StableDiffusion:
else:
Pipe = StableDiffusionAdapterPipeline
extra_args['adapter'] = self.adapter
else:
if self.is_xl:
extra_args['add_watermarker'] = False
# TODO add clip skip
if self.is_xl:
@@ -321,7 +324,6 @@ class StableDiffusion:
tokenizer=self.tokenizer[0],
tokenizer_2=self.tokenizer[1],
scheduler=noise_scheduler,
add_watermarker=False,
**extra_args
).to(self.device_torch)
# force turn that (ruin your images with obvious green and red dots) the #$@@ off!!!