From d7d6841406a2cef52da26fc58342e543b5cd9e1d Mon Sep 17 00:00:00 2001 From: Hu Ye Date: Wed, 26 Oct 2022 20:26:17 +0800 Subject: [PATCH] fix a bug in the new version (#957) remove tensor_format in the new version --- examples/text_to_image/train_text_to_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/text_to_image/train_text_to_image.py b/examples/text_to_image/train_text_to_image.py index c3c3c754..859a891e 100644 --- a/examples/text_to_image/train_text_to_image.py +++ b/examples/text_to_image/train_text_to_image.py @@ -375,7 +375,7 @@ def main(): # TODO (patil-suraj): load 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 ) # Get the datasets: you can either provide your own training and evaluation files (see below)