diff options
| author | Volpeon <git@volpeon.ink> | 2023-02-17 14:53:25 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-02-17 14:53:25 +0100 |
| commit | 842f26654bbe7dfd2f45df1fd2660d3f902af8cc (patch) | |
| tree | 3e7cd2dea37f025f9aa2755a893efd29195c7396 /training/functional.py | |
| parent | Fix (diff) | |
| download | textual-inversion-diff-842f26654bbe7dfd2f45df1fd2660d3f902af8cc.tar.gz textual-inversion-diff-842f26654bbe7dfd2f45df1fd2660d3f902af8cc.tar.bz2 textual-inversion-diff-842f26654bbe7dfd2f45df1fd2660d3f902af8cc.zip | |
Remove xformers, switch to Pytorch Nightly
Diffstat (limited to 'training/functional.py')
| -rw-r--r-- | training/functional.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/training/functional.py b/training/functional.py index 78a2b10..41794ea 100644 --- a/training/functional.py +++ b/training/functional.py | |||
| @@ -12,7 +12,7 @@ from torch.utils.data import DataLoader | |||
| 12 | 12 | ||
| 13 | from accelerate import Accelerator | 13 | from accelerate import Accelerator |
| 14 | from transformers import CLIPTextModel | 14 | from transformers import CLIPTextModel |
| 15 | from diffusers import AutoencoderKL, DDPMScheduler, UNet2DConditionModel | 15 | from diffusers import AutoencoderKL, DDPMScheduler, UNet2DConditionModel, UniPCMultistepScheduler |
| 16 | 16 | ||
| 17 | from tqdm.auto import tqdm | 17 | from tqdm.auto import tqdm |
| 18 | from PIL import Image | 18 | from PIL import Image |
| @@ -22,7 +22,6 @@ from pipelines.stable_diffusion.vlpn_stable_diffusion import VlpnStableDiffusion | |||
| 22 | from models.clip.embeddings import ManagedCLIPTextEmbeddings, patch_managed_embeddings | 22 | from models.clip.embeddings import ManagedCLIPTextEmbeddings, patch_managed_embeddings |
| 23 | from models.clip.util import get_extended_embeddings | 23 | from models.clip.util import get_extended_embeddings |
| 24 | from models.clip.tokenizer import MultiCLIPTokenizer | 24 | from models.clip.tokenizer import MultiCLIPTokenizer |
| 25 | from schedulers.scheduling_unipc_multistep import UniPCMultistepScheduler | ||
| 26 | from training.util import AverageMeter | 25 | from training.util import AverageMeter |
| 27 | 26 | ||
| 28 | 27 | ||