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