Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

print status message at startup when xformers is available

Browse Source
This commit is contained in:
Lincoln Stein 2023-01-30 22:01:06 -05:00
parent 371edc993a
commit b9cc56593e
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ldm/generate.py
View File

@ -20,6 +20,7 @@ import torch
import transformers import transformers
from PIL import Image, ImageOps from PIL import Image, ImageOps
from diffusers.pipeline_utils import DiffusionPipeline from diffusers.pipeline_utils import DiffusionPipeline
from diffusers.utils.import_utils import is_xformers_available
from omegaconf import OmegaConf from omegaconf import OmegaConf
from pytorch_lightning import seed_everything, logging from pytorch_lightning import seed_everything, logging
@ -203,6 +204,14 @@ class Generate:
self.precision = choose_precision(self.device) self.precision = choose_precision(self.device)
Globals.full_precision = self.precision=='float32' Globals.full_precision = self.precision=='float32'
if is_xformers_available():
if not Globals.disable_xformers:
print('>> xformers memory-efficient attention is available and enabled')
else:
print('>> xformers memory-efficient attention is available but disabled')
else:
print('>> xformers not installed')
# model caching system for fast switching # model caching system for fast switching
self.model_manager = ModelManager(mconfig,self.device,self.precision,max_loaded_models=max_loaded_models) self.model_manager = ModelManager(mconfig,self.device,self.precision,max_loaded_models=max_loaded_models)
# don't accept invalid models # don't accept invalid models