From 5b54788842cdd7b342bd60d6944158009130b4d4 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Wed, 28 Sep 2022 15:05:30 +0200 Subject: Improved sample output and progress bars --- textual_inversion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'textual_inversion.py') diff --git a/textual_inversion.py b/textual_inversion.py index 81f1cf5..399d876 100644 --- a/textual_inversion.py +++ b/textual_inversion.py @@ -581,7 +581,7 @@ def main(): # TODO (patil-suraj): laod scheduler using args noise_scheduler = DDPMScheduler( - beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000, tensor_format="pt" + beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000 ) datamodule = CSVDataModule( -- cgit v1.2.3-54-g00ecf