model volume rename, env for flyio, public and private volumes tested tgt, everything is a public model (#3)

This commit is contained in:
Nick Kao
2024-01-27 10:21:15 +08:00
committed by GitHub
parent 95e454e687
commit ac5aba2aa9
9 changed files with 1306 additions and 9 deletions
+1
View File
@@ -5,3 +5,4 @@ CIVITAI_API_KEY=
# On production set to False
DEPLOY_TEST_FLAG=True
CIVITAI_API_KEY=
PUBLIC_MODEL_VOLUME_NAME=
+3 -2
View File
@@ -47,7 +47,8 @@ machine_id_websocket_dict = {}
machine_id_status = {}
fly_instance_id = os.environ.get('FLY_ALLOC_ID', 'local').split('-')[0]
civitai_api_key = os.environ.get('FLY_ALLOC_ID', 'local').split('-')[0]
civitai_api_key = os.environ.get('FLY_ALLOC_ID', 'local')
public_model_volume_name = os.environ.get('PUBLIC_MODEL_VOLUME_NAME', 'local')
class FlyReplayMiddleware(BaseHTTPMiddleware):
@@ -388,7 +389,7 @@ async def build_logic(item: Item):
"name": item.name,
"deploy_test": os.environ.get("DEPLOY_TEST_FLAG", "False"),
"gpu": item.gpu,
"public_model_volume": "model-store",
"public_model_volume": public_model_volume_name,
"private_model_volume": item.model_volume_name,
"pip": list(pip_modules)
}