[examples/dreambooth] don't pass tensor_format to scheduler. (huggingface#649)

patil-suraj · Prathik Rao · commit 21602831e882 · 2022-10-26T15:10:04.000-07:00
don't pass tensor_format
diff --git 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(

Original file line number	Diff line number	Diff line change
`@@ -434,7 +434,7 @@ def main():`
`434`	`434`	`)`
`435`	`435`
`436`	`436`	`noise_scheduler = DDPMScheduler(`
`437`		`- beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000, tensor_format="pt"`
	`437`	`+ beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000`
`438`	`438`	`)`
`439`	`439`
`440`	`440`	`train_dataset = DreamBoothDataset(`