1 files changed, 1 insertions, 1 deletions

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(