Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "071807c853d28f04f8f9b1db54cd124fd6fcda2c"
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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/text_to_image/train_text_to_image.py
View file @ d7d68414
...@@ -375,7 +375,7 @@ def main(): ...@@ -375,7 +375,7 @@ def main():
# TODO (patil-suraj): load scheduler using args # TODO (patil-suraj): load scheduler using args
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
) )
# Get the datasets: you can either provide your own training and evaluation files (see below) # 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