Commit History

Update model file path in ModelManager; change from poeticagpt-quantized-new.pth to poeticagpt.pth
9e5e405

abhisheksan commited on

Fix logging issues in poetry_generation.log; add warnings for custom vocabulary loading and model file not found errors, and ensure BASE_MODEL_DIR ends with a slash
a96ed1b

abhisheksan commited on

Refactor logging setup in main.py; improve file handler configuration with fallback to stdout and update model path
09d9560

abhisheksan commited on

Add model files and tokenizer configuration; include special tokens and initialization settings
e945d50

abhisheksan commited on

Refactor GenerateRequest model; remove field aliases for emotional_tone, creative_style, language_variety, and word_repetition, and update validation error messages to match new field names
c346d0a

abhisheksan commited on

Refactor GenerateRequest model; update emotional_tone field alias and improve field definitions for clarity
b13f641

abhisheksan commited on

Refactor GenerateRequest model; remove unnecessary aliases for emotional_tone and word_repetition fields
c90fa16

abhisheksan commited on

Refactor GenerateRequest model; update field names to snake_case, add aliases, and improve validation methods
d8c637e

abhisheksan commited on

Enhance model initialization and download process; add directory verification, error handling, and temporary file management
1ccdaa7

abhisheksan commited on

Refactor model initialization logic; improve error handling and logging for better diagnostics
23e1d87

abhisheksan commited on

Implement model initialization and health check endpoints; add model download logic, error handling, and request validation
0d8e806

abhisheksan commited on

Add FastAPI app initialization in main.py
ad71c98

abhisheksan commited on

Refactor model path handling in main.py; define BASE_DIR and MODEL_DIR, ensure model directory exists, and set model download URL
63e6606

abhisheksan commited on

Fix import statement for MODEL_PATH in main.py to reflect updated module structure
8705f74

abhisheksan commited on

Enhance model initialization and error handling; implement model path verification, improve logging configuration, and modify startup event for better diagnostics
6d653b3

abhisheksan commited on

Update model configuration and enhance initialization logic; adjust BASE_DIR for container, implement model download functionality, and improve health check response
110ce35

abhisheksan commited on

Set permissions for /app directory in Dockerfile
4bb8b78

abhisheksan commited on

Fix import statement for MODEL_PATH in main.py
fba7988

abhisheksan commited on

Enhance Poetry Generator API; implement health check endpoint, improve model loading with logging, and update request/response models
67dd542

abhisheksan commited on

Refactor poem generation logic; implement lazy loading for model and update request/response models
11be554

abhisheksan commited on

Remove redundant model initialization parameters for improved clarity
2107ef1

abhisheksan commited on

Remove python-multipart dependency from requirements.txt
a1ff764

abhisheksan commited on

Update dependencies in requirements.txt; upgrade FastAPI and Uvicorn, add new libraries for model handling and API requests
e085617

abhisheksan commited on

Refactor FastAPI application structure; implement CORS middleware, update model initialization, and enhance poem generation endpoint with new request and response models
c9cdb74

abhisheksan commited on

Implement model preloading in PoetryGenerationService for faster startup; remove redundant preload method from ModelManager
18faacf

abhisheksan commited on

Remove unused import of PoetryGenerationService to clean up code
8ecf9f2

abhisheksan commited on

Refactor poem generation parameters; replace PoemRequest with GenerationParams for enhanced control and validation of generation settings
6d783aa

abhisheksan commited on

Remove temperature parameter from generate_poem function call for simplified text generation
1f5f1e2

abhisheksan commited on

Refactor ModelManager and PoetryGenerationService for CPU optimization; enhance model initialization, improve text generation parameters, and implement explicit resource cleanup
91fceab

abhisheksan commited on

Refactor PoetryGenerationService and ModelManager for CPU optimization; update model and tokenizer initialization, simplify prompt templates, and enhance text generation and cleaning processes
fc2af9b

abhisheksan commited on

Update generate_poem function to use max_new_tokens for improved token control
faba267

abhisheksan commited on

Refactor PoetryGenerationService to improve model preloading and poem generation; switch to max_new_tokens for better control and update model name for consistency
17244ba

abhisheksan commited on

Add model preloading functionality to PoetryGenerationService for improved startup efficiency and error handling
25289e0

abhisheksan commited on

Refactor PoetryGenerationService to enhance poem generation; switch to GPT2-medium model, improve prompt templates, and implement output cleaning for better formatting and quality
63ead34

abhisheksan commited on

Refactor pipeline initialization in ModelManager to use device_map for automatic device selection
e75cb89

abhisheksan commited on

Update model name in PoetryGenerationService to use distilbert/distilgpt2 for improved compatibility in low-resource environments
bb5b157

abhisheksan commited on

Refactor PoetryGenerationService and ModelManager for improved resource efficiency; optimize model loading, implement caching, and enhance error handling
7257855

abhisheksan commited on

Add bitsandbytes to requirements for enhanced model performance
a1e3bb4

abhisheksan commited on

Optimize model loading and error handling in PoetryGenerationService; implement async poem generation and enhance application startup process
6dbb459

abhisheksan commited on

Improve model preloading in PoetryGenerationService with meaningful return value and enhanced error handling
abc61cb

abhisheksan commited on

Refactor PoetryGenerationService to streamline model initialization and improve error handling
f55cd01

abhisheksan commited on

Set HF_HOME environment variable for caching in Dockerfile
86e94f2

abhisheksan commited on

Enhance lifespan management in FastAPI by initializing PoetryGenerationService and handling model preloading asynchronously
1c1ca6d

abhisheksan commited on

Refactor model loading to use a consistent model name variable in PoetryGenerationService
51ed73b

abhisheksan commited on

Fix variable name for model loading in PoetryGenerationService
34aad78

abhisheksan commited on

Add model preloading functionality to PoetryGenerationService and initialize on startup
e47369c

abhisheksan commited on

Update model name in ModelManager to Llama-3.2-1B-Instruct
5118191

abhisheksan commited on

Remove optimized module usage from poetry generation service
7b5bffa

abhisheksan commited on

Remove optimized module usage from poetry generation and downgrade PyTorch version to 2.3.1
ac829a5

abhisheksan commited on