Skip to content

Working on using phi-3 with PyTorch in multiple environments

License

Notifications You must be signed in to change notification settings

niznik-dev/phi-3-playground

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

phi-3-playground

Working on using Phi-3 with PyTorch in multiple environments

Local: MacBook Pro 16-inch 2023 w/ M2 Pro & 32 GB RAM (Typical Footprint: ~16 GB in RAM)
Remote: Adroit (Typical Footprint: ~9 GB in GPU)

Model Location

You'll have to update *_PHI_PATH to match wherever you've downloaded Phi-3

Environment

I'm using an Anaconda environment with:

  • python==3.11
  • torch==2.2.1 (-c pytorch -c nvidia)
  • transformers==4.40 (pip)

When using GPU:

  • accelerate==0.29.3 (pip)
  • flash-attn==2.5.8 (pip)

About

Working on using phi-3 with PyTorch in multiple environments

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published