Skip to content

ouhenio/cog-playground

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Cog Playground

I'm trying to get familiar with Cog, so this repo serves as a backup of my learning journey.

Todo

  • Build a simple GPT inference pipeline.
  • Build a streaming GPT pipeline.
  • Implement a Stable Diffusion pipeline.
    • Add tracing and model warmup.

Usage

GPT2

Basic inference:

cd nlp/basic
cog predict -i prompt="Hello!" -i max_length=100

Streaming inference:

cd nlp/streaming
cog predict -i prompt="Count form one to twenty: One," -i max_length=100

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages