diffusers no longer use tensor_format in DDPMScheduler

2022-11-05 11:05:45 +00:00 · 2022-11-05 11:05:45 +00:00 · 0e0acf31b1
parent 74ef98667c
commit 0e0acf31b1
1 changed files with 0 additions and 1 deletions
--- 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