AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

tensor format ort bug fix (#1557)

Browse Source 
bug fix

Co-authored-by: Prathik Rao <prathikrao@microsoft.com>
Co-authored-by: anton- <anton@huggingface.co>
This commit is contained in:
Prathik Rao
2022-12-12 04:56:02 -08:00
committed by GitHub
parent 589330595d
commit 4645e28355
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/unconditional_image_generation/train_unconditional_ort.py
View File

@@ -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,