mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
d1a2c4cd8c
* Implements rudimentary api * Fixes blocking in API * Adds UI to monorepo > src/frontend/ * Updates frontend/README * Reverts conda env name to `ldm` * Fixes environment yamls * CORS config for testing * Fixes LogViewer position * API WID * Adds actions to image viewer * Increases vite chunkSizeWarningLimit to 1500 * Implements init image * Implements state persistence in localStorage * Improve progress data handling * Final build * Fixes mimetypes error on windows * Adds error logging * Fixes bugged img2img strength component * Adds sourcemaps to dev build * Fixes missing key * Changes connection status indicator to text * Adds ability to serve other hosts than localhost * Adding Flask API server * Removes source maps from config * Fixes prop transfer * Add missing packages and add CORS support * Adding API doc * Remove defaults from openapi doc * Adds basic error handling for server config query * Mostly working socket.io implementation. * Fixes bug preventing mask upload * Fixes bug with sampler name not written to metadata * UI Overhaul, numerous fixes Co-authored-by: Kyle Schouviller <kyle0654@hotmail.com> Co-authored-by: Lincoln Stein <lincoln.stein@gmail.com>
48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
def parse_seed_weights(seed_weights):
|
|
"""
|
|
Accepts seed weights as string in "12345:0.1,23456:0.2,3456:0.3" format
|
|
Validates them
|
|
If valid: returns as [[12345, 0.1], [23456, 0.2], [3456, 0.3]]
|
|
If invalid: returns False
|
|
"""
|
|
|
|
# Must be a string
|
|
if not isinstance(seed_weights, str):
|
|
return False
|
|
# String must not be empty
|
|
if len(seed_weights) == 0:
|
|
return False
|
|
|
|
pairs = []
|
|
|
|
for pair in seed_weights.split(","):
|
|
split_values = pair.split(":")
|
|
|
|
# Seed and weight are required
|
|
if len(split_values) != 2:
|
|
return False
|
|
|
|
if len(split_values[0]) == 0 or len(split_values[1]) == 1:
|
|
return False
|
|
|
|
# Try casting the seed to int and weight to float
|
|
try:
|
|
seed = int(split_values[0])
|
|
weight = float(split_values[1])
|
|
except ValueError:
|
|
return False
|
|
|
|
# Seed must be 0 or above
|
|
if not seed >= 0:
|
|
return False
|
|
|
|
# Weight must be between 0 and 1
|
|
if not (weight >= 0 and weight <= 1):
|
|
return False
|
|
|
|
# This pair is valid
|
|
pairs.append([seed, weight])
|
|
|
|
# All pairs are valid
|
|
return pairs
|