From 0e0acf31b1b0b946f7b5a0a0298c33a74067762b Mon Sep 17 00:00:00 2001
From: Astralite Heart <astralite.heart@gmail.com>
Date: Sat, 5 Nov 2022 11:05:45 +0000
Subject: [PATCH] diffusers no longer use tensor_format in DDPMScheduler

---
 diffusers_trainer.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/diffusers_trainer.py b/diffusers_trainer.py
index 8631ccf..2233aaa 100644
--- a/diffusers_trainer.py
+++ b/diffusers_trainer.py
@@ -523,7 +523,6 @@ def main():
         beta_end=0.012,
         beta_schedule='scaled_linear',
         num_train_timesteps=1000,
-        tensor_format='pt'
     )
 
     # load dataset