mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
lora modularize code
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -41,10 +41,10 @@ def get_embeddings():
|
||||
return {"loaded": convert_embeddings(db.word_embeddings), "skipped": convert_embeddings(db.skipped_embeddings)}
|
||||
|
||||
def get_loras():
|
||||
from modules.lora import network, networks
|
||||
from modules.lora import network, lora_load
|
||||
def create_lora_json(obj: network.NetworkOnDisk):
|
||||
return { "name": obj.name, "alias": obj.alias, "path": obj.filename, "metadata": obj.metadata }
|
||||
return [create_lora_json(obj) for obj in networks.available_networks.values()]
|
||||
return [create_lora_json(obj) for obj in lora_load.available_networks.values()]
|
||||
|
||||
def get_extra_networks(page: Optional[str] = None, name: Optional[str] = None, filename: Optional[str] = None, title: Optional[str] = None, fullname: Optional[str] = None, hash: Optional[str] = None): # pylint: disable=redefined-builtin
|
||||
res = []
|
||||
@@ -134,8 +134,8 @@ def post_refresh_vae():
|
||||
return shared.refresh_vaes()
|
||||
|
||||
def post_refresh_loras():
|
||||
from modules.lora import networks
|
||||
return networks.list_available_networks()
|
||||
from modules.lora import lora_load
|
||||
return lora_load.list_available_networks()
|
||||
|
||||
def get_extensions_list():
|
||||
from modules import extensions
|
||||
|
||||
@@ -74,7 +74,7 @@ def register_api(app: FastAPI): # register api
|
||||
manager = ConnectionManager()
|
||||
|
||||
def get_video_thumbnail(filepath):
|
||||
from modules.ui_control_helpers import get_video_params
|
||||
from modules.video import get_video_params
|
||||
try:
|
||||
stat = os.stat(filepath)
|
||||
frames, fps, duration, width, height, codec, frame = get_video_params(filepath, capture=True)
|
||||
|
||||
Reference in New Issue
Block a user