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():
)
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(
......
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