mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 16:39:47 +00:00
Fixed issues with adapter and disabled telementry for now
This commit is contained in:
3
run.py
3
run.py
@@ -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
|
||||
|
||||
@@ -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!!!
|
||||
|
||||
Reference in New Issue
Block a user