remove unused schema, lint
This commit is contained in:
parent
db82c6b760
commit
a71298ff33
|
@ -16,7 +16,7 @@ logger = getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class BlendImg2ImgStage(BaseStage):
|
class BlendImg2ImgStage(BaseStage):
|
||||||
max_tile = SizeChart.unlimited
|
max_tile = SizeChart.max
|
||||||
|
|
||||||
def run(
|
def run(
|
||||||
self,
|
self,
|
||||||
|
|
|
@ -22,7 +22,7 @@ logger = getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class SourceTxt2ImgStage(BaseStage):
|
class SourceTxt2ImgStage(BaseStage):
|
||||||
max_tile = SizeChart.unlimited
|
max_tile = SizeChart.max
|
||||||
|
|
||||||
def run(
|
def run(
|
||||||
self,
|
self,
|
||||||
|
|
|
@ -24,7 +24,7 @@ logger = getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class UpscaleOutpaintStage(BaseStage):
|
class UpscaleOutpaintStage(BaseStage):
|
||||||
max_tile = SizeChart.unlimited
|
max_tile = SizeChart.max
|
||||||
|
|
||||||
def run(
|
def run(
|
||||||
self,
|
self,
|
||||||
|
|
|
@ -543,9 +543,6 @@ def blend_loras(
|
||||||
if len(unmatched_keys) > 0:
|
if len(unmatched_keys) > 0:
|
||||||
logger.warning("could not find nodes for some keys: %s", unmatched_keys)
|
logger.warning("could not find nodes for some keys: %s", unmatched_keys)
|
||||||
|
|
||||||
# if model_type == "unet":
|
|
||||||
# save_model(base_model, f"/tmp/lora_blend_{model_type}.onnx", save_as_external_data=True, all_tensors_to_one_file=True, location="weights.pb")
|
|
||||||
|
|
||||||
return base_model
|
return base_model
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -24,7 +24,7 @@ class SizeChart(IntEnum):
|
||||||
hd16k = 2**14
|
hd16k = 2**14
|
||||||
hd32k = 2**15
|
hd32k = 2**15
|
||||||
hd64k = 2**16
|
hd64k = 2**16
|
||||||
unlimited = 2**32 # sort of
|
max = 2**32 # should be a reasonable upper limit for now
|
||||||
|
|
||||||
|
|
||||||
class TileOrder:
|
class TileOrder:
|
||||||
|
|
|
@ -1,117 +0,0 @@
|
||||||
$id: TODO
|
|
||||||
$schema: https://json-schema.org/draft/2020-12/schema
|
|
||||||
|
|
||||||
$defs:
|
|
||||||
grid:
|
|
||||||
type: object
|
|
||||||
additionalProperties: False
|
|
||||||
required: [width, height]
|
|
||||||
width:
|
|
||||||
type: number
|
|
||||||
height:
|
|
||||||
type: number
|
|
||||||
labels:
|
|
||||||
type: object
|
|
||||||
additionalProperties: False
|
|
||||||
properties:
|
|
||||||
title:
|
|
||||||
type: string
|
|
||||||
rows:
|
|
||||||
type: array
|
|
||||||
items:
|
|
||||||
type: string
|
|
||||||
columns:
|
|
||||||
type: array
|
|
||||||
items:
|
|
||||||
type: string
|
|
||||||
order:
|
|
||||||
type: array
|
|
||||||
items: number
|
|
||||||
|
|
||||||
job_base:
|
|
||||||
type: object
|
|
||||||
additionalProperties: true
|
|
||||||
required: [
|
|
||||||
device,
|
|
||||||
model,
|
|
||||||
pipeline,
|
|
||||||
scheduler,
|
|
||||||
prompt,
|
|
||||||
cfg,
|
|
||||||
steps,
|
|
||||||
seed,
|
|
||||||
]
|
|
||||||
properties:
|
|
||||||
batch:
|
|
||||||
type: number
|
|
||||||
device:
|
|
||||||
type: string
|
|
||||||
model:
|
|
||||||
type: string
|
|
||||||
control:
|
|
||||||
type: string
|
|
||||||
pipeline:
|
|
||||||
type: string
|
|
||||||
scheduler:
|
|
||||||
type: string
|
|
||||||
prompt:
|
|
||||||
type: string
|
|
||||||
negative_prompt:
|
|
||||||
type: string
|
|
||||||
cfg:
|
|
||||||
type: number
|
|
||||||
eta:
|
|
||||||
type: number
|
|
||||||
steps:
|
|
||||||
type: number
|
|
||||||
tiled_vae:
|
|
||||||
type: boolean
|
|
||||||
tiles:
|
|
||||||
type: number
|
|
||||||
overlap:
|
|
||||||
type: number
|
|
||||||
seed:
|
|
||||||
type: number
|
|
||||||
stride:
|
|
||||||
type: number
|
|
||||||
|
|
||||||
job_txt2img:
|
|
||||||
allOf:
|
|
||||||
- $ref: "#/$defs/job_base"
|
|
||||||
- type: object
|
|
||||||
additionalProperties: False
|
|
||||||
required: [
|
|
||||||
height,
|
|
||||||
width,
|
|
||||||
]
|
|
||||||
properties:
|
|
||||||
width:
|
|
||||||
type: number
|
|
||||||
height:
|
|
||||||
type: number
|
|
||||||
|
|
||||||
job_img2img:
|
|
||||||
allOf:
|
|
||||||
- $ref: "#/$defs/job_base"
|
|
||||||
- type: object
|
|
||||||
additionalProperties: False
|
|
||||||
required: []
|
|
||||||
properties:
|
|
||||||
loopback:
|
|
||||||
type: number
|
|
||||||
|
|
||||||
type: object
|
|
||||||
additionalProperties: False
|
|
||||||
properties:
|
|
||||||
txt2img:
|
|
||||||
type: array
|
|
||||||
items:
|
|
||||||
$ref: "#/$defs/job_txt2img"
|
|
||||||
img2img:
|
|
||||||
type: array
|
|
||||||
items:
|
|
||||||
$ref: "#/$defs/job_img2img"
|
|
||||||
grid:
|
|
||||||
type: array
|
|
||||||
items:
|
|
||||||
$ref: "#/$defs/grid"
|
|
|
@ -0,0 +1,26 @@
|
||||||
|
import unittest
|
||||||
|
|
||||||
|
from onnx_web.chain.base import ChainProgress
|
||||||
|
|
||||||
|
|
||||||
|
class ChainProgressTests(unittest.TestCase):
|
||||||
|
def test_accumulate_with_reset(self):
|
||||||
|
def parent(step, timestep, latents):
|
||||||
|
pass
|
||||||
|
|
||||||
|
progress = ChainProgress(parent)
|
||||||
|
progress(5, 1, None)
|
||||||
|
progress(0, 1, None)
|
||||||
|
progress(5, 1, None)
|
||||||
|
|
||||||
|
self.assertEqual(progress.get_total(), 10)
|
||||||
|
|
||||||
|
def test_start_value(self):
|
||||||
|
def parent(step, timestep, latents):
|
||||||
|
pass
|
||||||
|
|
||||||
|
progress = ChainProgress(parent, 5)
|
||||||
|
self.assertEqual(progress.get_total(), 5)
|
||||||
|
|
||||||
|
progress(10, 1, None)
|
||||||
|
self.assertEqual(progress.get_total(), 10)
|
|
@ -150,8 +150,17 @@ class KernelSliceTests(unittest.TestCase):
|
||||||
(2, 2),
|
(2, 2),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
class BlendLoRATests(unittest.TestCase):
|
class BlendLoRATests(unittest.TestCase):
|
||||||
pass
|
def test_blend_unet(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def test_blend_text_encoder(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def test_blend_text_encoder_index(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
class InterpToMatchTests(unittest.TestCase):
|
class InterpToMatchTests(unittest.TestCase):
|
||||||
def test_same_shape(self):
|
def test_same_shape(self):
|
||||||
|
|
Loading…
Reference in New Issue