Unverified Commit ac665b64 authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

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

don't pass tensor_format
parent bd8df2da
...@@ -434,7 +434,7 @@ def main(): ...@@ -434,7 +434,7 @@ def main():
) )
noise_scheduler = DDPMScheduler( 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( train_dataset = DreamBoothDataset(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment