Skip to content

Pandora: Addition of Further AI and Ollama Integration#471

Merged
sei-dupdyke merged 5 commits into
cmu-sei:masterfrom
dh4340:pandora_image_gen
Dec 5, 2024
Merged

Pandora: Addition of Further AI and Ollama Integration#471
sei-dupdyke merged 5 commits into
cmu-sei:masterfrom
dh4340:pandora_image_gen

Conversation

@dh4340

@dh4340 dh4340 commented Nov 28, 2024

Copy link
Copy Markdown
Contributor

Description

This PR introduces enhancements to the project by incorporating further AI capabilities and deeper integration with Ollama, along with additional improvements for maintainability and performance.


Key Improvements

  1. Centralised Configuration:

    • Consolidated configuration settings into a unified structure for improved maintainability and consistency.
    • Eliminated redundancy across modules by leveraging shared configuration components.
  2. Enhanced Logging:

    • Transitioned to JSON-based logging for better compatibility with modern logging ingestion systems.
    • Standardised all log outputs with f-strings for clarity and uniformity.
    • Added multi-level logging throughout the application for enhanced traceability and debugging.
  3. Optimised Models:

    • Integrated smaller and more generic models to improve response quality and efficiency.
    • These models enhance resource usage while maintaining high-quality outputs.
  4. Image Generation:

    • Migrated to ImaginAIry for generating images, aligning with shadow configurations and improving image quality.
  5. Extended Ollama Integration:

    • Applied Ollama for document and dialogue generation across multiple API endpoints, broadening its application scope.

New Features

  1. Video Generation API:

    • Added a short video generation API capable of producing professional-quality videos.
    • Designed for optimal performance on H100 or H200 GPUs (Hopper architecture).
    • Supports GPUs with Ampere architecture, with reduced memory requirements.
  2. Voice Generation API:

    • Added support for realistic conversation simulation.
    • Dynamically generates voice dialogues based on user-defined actor roles and contexts.

Technical Highlights

  • Video Generation:

    • Endpoint: /video/{file_name}
    • Supports generating realistic short videos, leveraging advanced AI capabilities.
    • GPU optimised, requiring Hopper or Ampere architecture.
  • Voice Generation:

    • Endpoint: /call
    • Facilitates synthesised conversations, allowing user-defined or default actor roles.
  • Dialogue Generation:

    • Enhanced formatting and consistency through Ollama, now standardised across video and voice APIs.
  • Compatibility Enhancements:

    • JSON logging ensures seamless integration with logging ingestion systems.
    • Migrating image generation to ImaginAIry aligns better with shadowed configurations.

Impact

  • Performance: Improved response times through optimised models and configurations.
  • Maintainability: Centralised settings and standardised logging streamline future enhancements.
  • Functionality: Expanded capabilities with new APIs for video and voice generation.
  • Scalability: Enhanced design ensures better utilisation of hardware resources, especially modern GPUs.

Next Steps

  • Implement unit tests for:
    • All endpoints.
    • Centralised configuration validation.

Checklist

  • Centralised configuration updates.
  • JSON logging integration.
  • Migration to ImaginAIry for image generation.
  • Video generation endpoint added.
  • Voice generation endpoint added.

Ready for review and feedback. Let me know if additional adjustments are required!

@sei-dupdyke
sei-dupdyke merged commit bfae266 into cmu-sei:master Dec 5, 2024
@dh4340
dh4340 deleted the pandora_image_gen branch December 6, 2024 10:57
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants