From d2c1103e5e52d1648f04087f4d4323a7adee29c2 Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Tue, 27 Sep 2022 17:22:54 +0200 Subject: [PATCH] don't pass tensor_format --- examples/dreambooth/train_dreambooth.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/dreambooth/train_dreambooth.py b/examples/dreambooth/train_dreambooth.py index 600653187977..3d4c23a4affa 100644 --- a/examples/dreambooth/train_dreambooth.py +++ b/examples/dreambooth/train_dreambooth.py @@ -434,7 +434,7 @@ def main(): ) 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 ) train_dataset = DreamBoothDataset(