tensor format ort bug fix (#1557)

prathikr · Prathik Rao · anton-l · web-flow · commit 4645e2835536 · 2022-12-12T13:56:02.000+01:00
bug fix

Co-authored-by: Prathik Rao &lt;prathikrao@microsoft.com&gt;
Co-authored-by: anton- &lt;anton@huggingface.co&gt;
diff --git a/examples/unconditional_image_generation/train_unconditional_ort.py b/examples/unconditional_image_generation/train_unconditional_ort.py
@@ -77,7 +77,7 @@ def main(args):
         ),
     )
     model = ORTModule(model)
-    noise_scheduler = DDPMScheduler(num_train_timesteps=1000, tensor_format="pt")
+    noise_scheduler = DDPMScheduler(num_train_timesteps=1000)
     optimizer = torch.optim.AdamW(
         model.parameters(),
         lr=args.learning_rate,

Original file line number	Diff line number	Diff line change
`@@ -77,7 +77,7 @@ def main(args):`
`77`	`77`	`),`
`78`	`78`	`)`
`79`	`79`	`model = ORTModule(model)`
`80`		`- noise_scheduler = DDPMScheduler(num_train_timesteps=1000, tensor_format="pt")`
	`80`	`+ noise_scheduler = DDPMScheduler(num_train_timesteps=1000)`
`81`	`81`	`optimizer = torch.optim.AdamW(`
`82`	`82`	`model.parameters(),`
`83`	`83`	`lr=args.learning_rate,`