Pin requirements for numpy and torch

pull/117/head
SKi 2 years ago
parent 6b55440312
commit ab3564b8d3
No known key found for this signature in database

@ -508,7 +508,7 @@ if __name__ == "__main__":
help="Random seed to be able to reproduce results") help="Random seed to be able to reproduce results")
parser.add_argument( parser.add_argument(
"--model-version", "--model-version",
default="CompVis/stable-diffusion-v1-4", default="stabilityai/stable-diffusion-2-base",
help= help=
("The pre-trained model checkpoint and configuration to restore. " ("The pre-trained model checkpoint and configuration to restore. "
"For available versions: https://huggingface.co/models?search=stable-diffusion" "For available versions: https://huggingface.co/models?search=stable-diffusion"

Loading…
Cancel
Save