2023-11-12 21:36:51 +00:00
|
|
|
from typing import List, Literal
|
2023-03-19 00:14:24 +00:00
|
|
|
|
2023-12-03 18:53:50 +00:00
|
|
|
NetworkType = Literal["control", "inversion", "lora"]
|
2023-03-19 00:14:24 +00:00
|
|
|
|
|
|
|
|
|
|
|
class NetworkModel:
|
|
|
|
name: str
|
2023-11-12 21:36:51 +00:00
|
|
|
tokens: List[str]
|
2023-03-19 00:14:24 +00:00
|
|
|
type: NetworkType
|
|
|
|
|
2023-11-12 22:38:56 +00:00
|
|
|
def __init__(self, name: str, type: NetworkType, tokens=None) -> None:
|
2023-03-19 00:14:24 +00:00
|
|
|
self.name = name
|
2023-11-12 21:36:51 +00:00
|
|
|
self.tokens = tokens or []
|
2023-03-19 00:14:24 +00:00
|
|
|
self.type = type
|
|
|
|
|
|
|
|
def tojson(self):
|
|
|
|
return {
|
|
|
|
"name": self.name,
|
2023-11-12 21:36:51 +00:00
|
|
|
"tokens": self.tokens,
|
2023-03-19 00:14:24 +00:00
|
|
|
"type": self.type,
|
|
|
|
}
|