rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
ebolam
2022-09-20 12:14:38 -04:00
parent 0819889095
commit d395465396
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
aiserver.py
View File

@@ -1209,7 +1209,6 @@ def general_startup(override_args=None):
parser.add_argument("--path", help="Specify the Path for local models (For model NeoCustom or GPT2Custom)") parser.add_argument("--path", help="Specify the Path for local models (For model NeoCustom or GPT2Custom)")
parser.add_argument("--apikey", help="Specify the API key to use for online services") parser.add_argument("--apikey", help="Specify the API key to use for online services")
parser.add_argument("--sh_apikey", help="Specify the API key to use for txt2img from the Stable Horde. Get a key from https://stablehorde.net/register") parser.add_argument("--sh_apikey", help="Specify the API key to use for txt2img from the Stable Horde. Get a key from https://stablehorde.net/register")
parser.add_argument("--text2img", action='store_true', default=False, help="Will convert the sent prompt into an image, using Stable Horde")
parser.add_argument("--req_model", type=str, action='append', required=False, help="Which models which we allow to generate for us during cluster mode. Can be specified multiple times.") parser.add_argument("--req_model", type=str, action='append', required=False, help="Which models which we allow to generate for us during cluster mode. Can be specified multiple times.")
parser.add_argument("--revision", help="Specify the model revision for huggingface models (can be a git branch/tag name or a git commit hash)") parser.add_argument("--revision", help="Specify the model revision for huggingface models (can be a git branch/tag name or a git commit hash)")
parser.add_argument("--cpu", action='store_true', help="By default unattended launches are on the GPU use this option to force CPU usage.") parser.add_argument("--cpu", action='store_true', help="By default unattended launches are on the GPU use this option to force CPU usage.")
@@ -1289,7 +1288,6 @@ def general_startup(override_args=None):
koboldai_vars.apikey = args.apikey koboldai_vars.apikey = args.apikey
if args.sh_apikey: if args.sh_apikey:
koboldai_vars.sh_apikey = args.sh_apikey koboldai_vars.sh_apikey = args.sh_apikey
koboldai_vars.text2img = args.text2img
if args.req_model: if args.req_model:
koboldai_vars.cluster_requested_models = args.req_model koboldai_vars.cluster_requested_models = args.req_model
@@ -8185,7 +8183,7 @@ def UI_2_generate_image(data):
#If we have > 4 keys, use those otherwise use sumarization #If we have > 4 keys, use those otherwise use sumarization
if len(keys) < 4: if len(keys) < 4:
from transformers import pipeline as summary_pipeline from transformers import pipeline as summary_pipeline
summarizer = summary_pipeline("summarization", model="sshleifer/distilbart-cnn-12-1") summarizer = summary_pipeline("summarization", model="sshleifer/distilbart-xsum-12-1")
#text to summarize: #text to summarize:
if len(koboldai_vars.actions) < 5: if len(koboldai_vars.actions) < 5:
text = "".join(koboldai_vars.actions[:-5]+[koboldai_vars.prompt]) text = "".join(koboldai_vars.actions[:-5]+[koboldai_vars.prompt])
@@ -8249,7 +8247,7 @@ def text2img_local(prompt, art_guide="", filename="new.png"):
koboldai_vars.image_pipeline = None koboldai_vars.image_pipeline = None
del pipe del pipe
koboldai_vars.generating_image = False koboldai_vars.generating_image = False
koboldai_vars.aibusy = True koboldai_vars.aibusy = False
print("time to unload: {}".format(time.time() - start_time)) print("time to unload: {}".format(time.time() - start_time))
return img_str return img_str