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

patil-suraj · web-flow · commit ac665b6484d9 · 2022-09-27T17:24:12.000+02: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(`