"docs/vscode:/vscode.git/clone" did not exist on "f6e2f05b16c1d13f0e017450cad975244099de80"
Unverified Commit d7d68414 authored by Hu Ye's avatar Hu Ye Committed by GitHub
Browse files

fix a bug in the new version (#957)

remove tensor_format in the new version
parent d9cfe325
......@@ -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)
......
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