19 lines
354 B
Python
19 lines
354 B
Python
|
from typing import Literal
|
||
|
|
|
||
|
|
NetworkType = Literal["inversion", "lora"]
|
||
|
|
|
||
|
|
|
||
|
|
class NetworkModel:
|
||
|
|
name: str
|
||
|
|
type: NetworkType
|
||
|
|
|
||
|
|
def __init__(self, name: str, type: NetworkType) -> None:
|
||
|
|
self.name = name
|
||
|
|
self.type = type
|
||
|
|
|
||
|
|
def tojson(self):
|
||
|
|
return {
|
||
|
|
"name": self.name,
|
||
|
|
"type": self.type,
|
||
|
|
}
|