Update dream.py. k_euler_a and k_dpm_2_a M1 fix (#579)

* Update dream.py. k_euler_a and k_dpm_2_a M1 fix

Make results reproducible (so runs with the same seed produce the same result).
Implements fix by @wbowling referenced in https://github.com/lstein/stable-diffusion/issues/397#issuecomment-1240679294

* Update dream.py. Remove import torch from dream.py

* generate.py: k_euler_a and k_dpm_2_a M1 fix #579

Co-authored-by: Lincoln Stein <lincoln.stein@gmail.com>
This commit is contained in:
Any-Winter-4079
2022-09-15 17:02:17 +02:00
committed by GitHub
parent ec2dc24ad7
commit 60b731e7ab
2 changed files with 18 additions and 1 deletions

View File

@ -20,7 +20,6 @@ from omegaconf import OmegaConf
# Just want to get the formatting look right for now.
output_cntr = 0
def main():
"""Initialize command-line parsers and the diffusion model"""
arg_parser = create_argv_parser()