Skip to content

isangtao/rtxailab

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

96 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RTX AI LAB

Screenshots

RTX AI Lab Screenshot

Objectives

  • to provide an environment where researchers can easily experiment with, compare and combine models to innovate quickly.
  • to provide minimalist source code for developers to copy and chain together in their own applications.
  • Please see LLM_POC_Overview_1.0.pdf for details

Description

  • This application combines the following modes into one UI.
    • Model comparison
    • Internet Search - RAG1
    • Personas & Living FAQs
    • Real-time Image Generation
    • Vision
    • Voice Cloning
  • This application has been developed and tested on Ubuntu 22.04 with an RTX4070 12GB and 32GB of RAM. This has not been tested on Windows or Mac.

Setup:

  • Download rtxailab.py
  • Install CUDA
  • Install MiniConda
  • Create and activate a new python environment
    • conda create -n rtxai python=3.9
    • conda activate rtxai
  • Install dependencies
    • pip install streamlit ollama openai duckduckgo_search streamlit-keyup diffusers torch transformers tts
    • You may need to exit your terminal, log back in and reactivate the rtxailab environment to register streamlit for the first time.
  • Install Ollama
    • Ref: https://ollama.com/
    • Ollama is available for Linux, Mac and Windows and supports the OpenAI API standard. The Linux install command is shown below.
    • curl -fsSL https://ollama.com/install.sh | sh
  • Download local LLMs
    • ollama pull llama3.2:latest
    • ollama pull llama3.2:1b
    • ollama pull qwen2.5-coder:latest
    • ollama pull qwen2.5-coder:1.5b
    • ollama pull moondream
    • Feel free to add any new models to the code.
  • Run application
    • streamlit run rtxailab.py

About

RTX AI Lab

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published