model volume rename, env for flyio, public and private volumes tested tgt, everything is a public model (#3)
This commit is contained in:
@@ -5,3 +5,4 @@ CIVITAI_API_KEY=
|
||||
# On production set to False
|
||||
DEPLOY_TEST_FLAG=True
|
||||
CIVITAI_API_KEY=
|
||||
PUBLIC_MODEL_VOLUME_NAME=
|
||||
|
||||
@@ -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)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user