fix a bug in the new version (#957)

xiaohu2015 · web-flow · commit d7d6841406a2 · 2022-10-26T14:26:17.000+02:00
remove tensor_format in the new version
diff --git a/examples/text_to_image/train_text_to_image.py b/examples/text_to_image/train_text_to_image.py
@@ -375,7 +375,7 @@ def main():
 
     # TODO (patil-suraj): load scheduler using args
     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
     )
 
     # Get the datasets: you can either provide your own training and evaluation files (see below)

Original file line number	Diff line number	Diff line change
`@@ -375,7 +375,7 @@ def main():`
`375`	`375`
`376`	`376`	`# TODO (patil-suraj): load scheduler using args`
`377`	`377`	`noise_scheduler = DDPMScheduler(`
`378`		`- beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000, tensor_format="pt"`
	`378`	`+ beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000`
`379`	`379`	`)`
`380`	`380`
`381`	`381`	`# Get the datasets: you can either provide your own training and evaluation files (see below)`