@@ -20,7 +20,7 @@ clip_model_name = 'ViT-L/14'
|
|||||||
|
|
||||||
Category = namedtuple("Category", ["name", "topn", "items"])
|
Category = namedtuple("Category", ["name", "topn", "items"])
|
||||||
|
|
||||||
re_topn = re.compile(r"\.top(\d+)\.")
|
re_topn = re.compile(r"\.top(\d+)$")
|
||||||
|
|
||||||
def category_types():
|
def category_types():
|
||||||
return [f.stem for f in Path(shared.interrogator.content_dir).glob('*.txt')]
|
return [f.stem for f in Path(shared.interrogator.content_dir).glob('*.txt')]
|
||||||
|
|||||||
+2
-2
@@ -60,10 +60,10 @@ def txt2img_upscale(id_task: str, request: gr.Request, gallery, gallery_index, g
|
|||||||
assert len(gallery) > 0, 'No image to upscale'
|
assert len(gallery) > 0, 'No image to upscale'
|
||||||
assert 0 <= gallery_index < len(gallery), f'Bad image index: {gallery_index}'
|
assert 0 <= gallery_index < len(gallery), f'Bad image index: {gallery_index}'
|
||||||
|
|
||||||
p = txt2img_create_processing(id_task, request, *args)
|
p = txt2img_create_processing(id_task, request, *args, force_enable_hr=True)
|
||||||
p.enable_hr = True
|
|
||||||
p.batch_size = 1
|
p.batch_size = 1
|
||||||
p.n_iter = 1
|
p.n_iter = 1
|
||||||
|
# txt2img_upscale attribute that signifies this is called by txt2img_upscale
|
||||||
p.txt2img_upscale = True
|
p.txt2img_upscale = True
|
||||||
|
|
||||||
geninfo = json.loads(generation_info)
|
geninfo = json.loads(generation_info)
|
||||||
|
|||||||
Reference in New Issue
Block a user