Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

remove load ds #318

Merged
merged 3 commits into from
Jul 25, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion backend/src/api/endpoints.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,4 +35,4 @@
router.include_router(router=version_router)
router.include_router(router=health_router)
router.include_router(router=datasets_router)
router.include_router(router=neural_nets_router)
router.include_router(router=neural_nets_router)
9 changes: 4 additions & 5 deletions backend/src/api/routes/neural_nets.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,6 @@
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/api/auth/verify")



@router.get(
path="/list",
name="nn:get-neural-net-list",
Expand All @@ -40,11 +39,11 @@ async def get_neural_net_list(
{
"name": "GPT",
"description": "GPT is a transformer-based model",
"modules": ["BigramLanguageModel((token_embedding_table): Embedding(65, 65))","Embedding(65, 65)"]
"modules": ["BigramLanguageModel((token_embedding_table): Embedding(65, 65))", "Embedding(65, 65)"],
},
{
"name": "Makemore",
"description": "Makemore is a character-level language model",
"modules": ["Makemore((token_embedding_table): Embedding(65, 65))","Embedding(65, 65)"]
}
]
"modules": ["Makemore((token_embedding_table): Embedding(65, 65))", "Embedding(65, 65)"],
},
]
2 changes: 1 addition & 1 deletion backend/src/models/schemas/neural_nets.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,4 +23,4 @@
class NeuralnetsResponse(BaseSchemaModel):
name: str = Field(..., title="Neural Network Name", description="Neural Network Name")
description: Optional[str] = Field(default="Neural nets", title="Description", description="Description")
modules: Optional[list[str]] = Field(default=["Embedding(65, 65)"], title="Modules", description="Modules")
modules: Optional[list[str]] = Field(default=["Embedding(65, 65)"], title="Modules", description="Modules")
14 changes: 0 additions & 14 deletions backend/src/repository/events.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,6 @@
from src.config.settings.const import ANONYMOUS_USER, ANONYMOUS_EMAIL, ANONYMOUS_PASS
from src.config.manager import settings
from src.models.db.account import Account
from src.models.db.dataset import DataSet
from src.securities.hashing.password import pwd_generator
from src.repository.database import async_db
from src.repository.table import Base
Expand Down Expand Up @@ -96,18 +95,6 @@ async def initialize_admin_user(async_session: AsyncSession) -> None:
loguru.logger.info("Admin user --- Successfully Created!")


async def initialize_rag_datasets(async_session: AsyncSession) -> None:
"""
Initialize ds for RAG
"""
loguru.logger.info("RAG datasets --- Creating . . .")
ds = DataSet(name=settings.DEFAULT_RAG_DS_NAME)
async_session.add(instance=ds)
await async_session.commit()
await async_session.refresh(instance=ds)
loguru.logger.info("RAG datasets --- Successfully Created!")


async def initialize_db_connection(backend_app: fastapi.FastAPI) -> None:
loguru.logger.info("Database Connection --- Establishing . . .")

Expand All @@ -118,7 +105,6 @@ async def initialize_db_connection(backend_app: fastapi.FastAPI) -> None:
async with async_db.async_session as async_session:
await initialize_anonymous_user(async_session=async_session)
await initialize_admin_user(async_session=async_session)
await initialize_rag_datasets(async_session=async_session)

loguru.logger.info("Database Connection --- Successfully Established!")

Expand Down
22 changes: 11 additions & 11 deletions docker-compose.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -198,17 +198,17 @@ services:
- kirin
- yeager

yeager:
container_name: yeager
image: ghcr.io/skywardai/yeager:v0.1.0
restart: always
entrypoint: ["python3", "-m", "tensorboard.main","--port=80","--bind_all","--logdir=${METRICS_PATHS}"]
volumes:
- ${DOCKER_VOLUME_DIRECTORY:-.}/volumes/runs:/app/runs
expose:
- 80
ports:
- 6006:80
# yeager:
# container_name: yeager
# image: ghcr.io/skywardai/yeager:v0.1.0
# restart: always
# entrypoint: ["python3", "-m", "tensorboard.main","--port=80","--bind_all","--logdir=${METRICS_PATHS}"]
# volumes:
# - ${DOCKER_VOLUME_DIRECTORY:-.}/volumes/runs:/app/runs
# expose:
# - 80
# ports:
# - 6006:80


volumes:
Expand Down