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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

149
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,7 +31,8 @@ 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"
setup_readline() if readline_available:
setup_readline()
print("* Initializing, be patient...\n") print("* Initializing, be patient...\n")
from pytorch_lightning import logging from pytorch_lightning import logging
@ -174,80 +180,81 @@ 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
def setup_readline(): if readline_available:
readline.set_completer(Completer(['--steps','-s','--seed','-S','--iterations','-n','--batch_size','-b', def setup_readline():
'--width','-W','--height','-H','--cfg_scale','-C','--grid','-g', readline.set_completer(Completer(['--steps','-s','--seed','-S','--iterations','-n','--batch_size','-b',
'--individual','-i','--init_img','-I','--strength','-f']).complete) '--width','-W','--height','-H','--cfg_scale','-C','--grid','-g',
readline.set_completer_delims(" ") '--individual','-i','--init_img','-I','--strength','-f']).complete)
readline.parse_and_bind('tab: complete') readline.set_completer_delims(" ")
load_history() readline.parse_and_bind('tab: complete')
load_history()
def load_history(): def load_history():
histfile = os.path.join(os.path.expanduser('~'),".dream_history") histfile = os.path.join(os.path.expanduser('~'),".dream_history")
try: try:
readline.read_history_file(histfile) readline.read_history_file(histfile)
readline.set_history_length(1000) readline.set_history_length(1000)
except FileNotFoundError: except FileNotFoundError:
pass pass
atexit.register(readline.write_history_file,histfile) atexit.register(readline.write_history_file,histfile)
class Completer(): class Completer():
def __init__(self,options): def __init__(self,options):
self.options = sorted(options) self.options = sorted(options)
return return
def complete(self,text,state): def complete(self,text,state):
if text.startswith('-I') or text.startswith('--init_img'): if text.startswith('-I') or text.startswith('--init_img'):
return self._image_completions(text,state) return self._image_completions(text,state)
response = None response = None
if state == 0: if state == 0:
# This is the first time for this text, so build a match list. # This is the first time for this text, so build a match list.
if text: if text:
self.matches = [s self.matches = [s
for s in self.options for s in self.options
if s and s.startswith(text)] if s and s.startswith(text)]
else:
self.matches = self.options[:]
# Return the state'th item from the match list,
# if we have that many.
try:
response = self.matches[state]
except IndexError:
response = None
return response
def _image_completions(self,text,state):
# get the path so far
if text.startswith('-I'):
path = text.replace('-I','',1).lstrip()
elif text.startswith('--init_img='):
path = text.replace('--init_img=','',1).lstrip()
matches = list()
path = os.path.expanduser(path)
if len(path)==0:
matches.append(text+'./')
else: else:
self.matches = self.options[:] dir = os.path.dirname(path)
dir_list = os.listdir(dir)
for n in dir_list:
if n.startswith('.') and len(n)>1:
continue
full_path = os.path.join(dir,n)
if full_path.startswith(path):
if os.path.isdir(full_path):
matches.append(os.path.join(os.path.dirname(text),n)+'/')
elif n.endswith('.png'):
matches.append(os.path.join(os.path.dirname(text),n))
# Return the state'th item from the match list, try:
# if we have that many. response = matches[state]
try: except IndexError:
response = self.matches[state] response = None
except IndexError: return response
response = None
return response
def _image_completions(self,text,state):
# get the path so far
if text.startswith('-I'):
path = text.replace('-I','',1).lstrip()
elif text.startswith('--init_img='):
path = text.replace('--init_img=','',1).lstrip()
matches = list()
path = os.path.expanduser(path)
if len(path)==0:
matches.append(text+'./')
else:
dir = os.path.dirname(path)
dir_list = os.listdir(dir)
for n in dir_list:
if n.startswith('.') and len(n)>1:
continue
full_path = os.path.join(dir,n)
if full_path.startswith(path):
if os.path.isdir(full_path):
matches.append(os.path.join(os.path.dirname(text),n)+'/')
elif n.endswith('.png'):
matches.append(os.path.join(os.path.dirname(text),n))
try:
response = matches[state]
except IndexError:
response = None
return response
if __name__ == "__main__": if __name__ == "__main__":