diff options
Diffstat (limited to 'textual_inversion.py')
-rw-r--r-- | textual_inversion.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/textual_inversion.py b/textual_inversion.py index bcdfd3a..dd7c3bd 100644 --- a/textual_inversion.py +++ b/textual_inversion.py | |||
@@ -22,7 +22,7 @@ from tqdm.auto import tqdm | |||
22 | from transformers import CLIPTextModel, CLIPTokenizer | 22 | from transformers import CLIPTextModel, CLIPTokenizer |
23 | from slugify import slugify | 23 | from slugify import slugify |
24 | 24 | ||
25 | from schedulers.scheduling_euler_a import EulerAScheduler | 25 | from schedulers.scheduling_euler_ancestral_discrete import EulerAncestralDiscreteScheduler |
26 | from pipelines.stable_diffusion.vlpn_stable_diffusion import VlpnStableDiffusion | 26 | from pipelines.stable_diffusion.vlpn_stable_diffusion import VlpnStableDiffusion |
27 | from data.csv import CSVDataModule | 27 | from data.csv import CSVDataModule |
28 | from models.clip.prompt import PromptProcessor | 28 | from models.clip.prompt import PromptProcessor |
@@ -398,7 +398,7 @@ class Checkpointer: | |||
398 | samples_path = Path(self.output_dir).joinpath("samples") | 398 | samples_path = Path(self.output_dir).joinpath("samples") |
399 | 399 | ||
400 | unwrapped = self.accelerator.unwrap_model(self.text_encoder) | 400 | unwrapped = self.accelerator.unwrap_model(self.text_encoder) |
401 | scheduler = EulerAScheduler( | 401 | scheduler = EulerAncestralDiscreteScheduler( |
402 | beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear" | 402 | beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear" |
403 | ) | 403 | ) |
404 | 404 | ||
@@ -639,7 +639,7 @@ def main(): | |||
639 | batched_data = [missing_data[i:i+args.sample_batch_size] | 639 | batched_data = [missing_data[i:i+args.sample_batch_size] |
640 | for i in range(0, len(missing_data), args.sample_batch_size)] | 640 | for i in range(0, len(missing_data), args.sample_batch_size)] |
641 | 641 | ||
642 | scheduler = EulerAScheduler( | 642 | scheduler = EulerAncestralDiscreteScheduler( |
643 | beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear" | 643 | beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear" |
644 | ) | 644 | ) |
645 | 645 | ||