Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

disable readline functionality on windows

Browse Source
This commit is contained in:
Lincoln Stein
2022-08-18 16:00:44 -04:00
parent bb46c70ec5
commit ac440a1197
1 changed files with 79 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
scripts/dream.py
View File

@ -1,11 +1,16 @@
#!/usr/bin/env python #!/usr/bin/env python
import readline
import argparse import argparse
import shlex import shlex
import atexit import atexit
import os import os
# readline will be disabled on windows systems
try:
import readline
readline_available = True
except:
readline_available = False
debugging = False debugging = False
def main(): def main():
@ -26,6 +31,7 @@ def main():
weights = "models/ldm/stable-diffusion-v1/model.ckpt" weights = "models/ldm/stable-diffusion-v1/model.ckpt"
# command line history will be stored in a file called "~/.dream_history" # command line history will be stored in a file called "~/.dream_history"
if readline_available:
setup_readline() setup_readline()
print("* Initializing, be patient...\n") print("* Initializing, be patient...\n")
@ -174,6 +180,7 @@ def create_cmd_parser():
parser.add_argument('-f','--strength',default=0.75,type=float,help="strength for noising/unnoising. 0.0 preserves image exactly, 1.0 replaces it completely") parser.add_argument('-f','--strength',default=0.75,type=float,help="strength for noising/unnoising. 0.0 preserves image exactly, 1.0 replaces it completely")
return parser return parser
if readline_available:
def setup_readline(): def setup_readline():
readline.set_completer(Completer(['--steps','-s','--seed','-S','--iterations','-n','--batch_size','-b', readline.set_completer(Completer(['--steps','-s','--seed','-S','--iterations','-n','--batch_size','-b',
'--width','-W','--height','-H','--cfg_scale','-C','--grid','-g', '--width','-W','--height','-H','--cfg_scale','-C','--grid','-g',