cleanup and update todo

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-04-29 12:19:03 -04:00
parent c940f69b6d
commit 52b6111b23
3 changed files with 2 additions and 16 deletions
-1
View File
@@ -68,7 +68,6 @@ def parse(infotext):
debug(f'Raw: {infotext}')
remaining = infotext.replace('\nSteps:', ' Steps:')
# TODO infotext: handle using regex instead
params = [' steps:', ' seed:', ' width:', ' height:', ' sampler:', ' size:', ' cfg scale:'] # first param is one of those
params += ['\nsteps:', '\nseed:', '\nwidth:', '\nheight:', '\nsampler:', '\nsize:', '\ncfg scale:']
params += ['.steps:', '.seed:', '.width:', '.height:', '.sampler:', '.size:', '.cfg scale:']
-10
View File
@@ -28,13 +28,6 @@ dtypes = {
}
def h1_pack_latents(latents, _batch_size, _num_channels_latents, _height, _width): # TODO hidream: pack latents for remote vae
# latents = latents.view(batch_size, num_channels_latents, height // 2, 2, width // 2, 2)
# latents = latents.permute(0, 2, 4, 1, 3, 5)
# latents = latents.reshape(batch_size, (height // 2) * (width // 2) // (num_channels_latents * 4), num_channels_latents * 4)
return latents
def remote_decode(latents: torch.Tensor, width: int = 0, height: int = 0, model_type: str = None) -> Image.Image:
from modules import devices, shared, errors, modelloader
tensors = []
@@ -57,9 +50,6 @@ def remote_decode(latents: torch.Tensor, width: int = 0, height: int = 0, model_
latent = latent_copy[i]
if model_type != 'f1':
latent = latent.unsqueeze(0)
# if model_type == 'h1':
# num_channels_latents = shared.sd_model.transformer.config.in_channels
# latent = h1_pack_latents(latent, 1, num_channels_latents, height, width) # pylint: disable=protected-access
params = {
"input_tensor_type": "binary",
"shape": list(latent.shape),