Files
AMD-SHARK-Studio/apps/language_models
Quinn Dawkins ded74d09cd [vicuna.py] Keep past key values on device (#1836)
The past key values are only used within the models themselves and can
be kept on device. For vulkan int4, this gives 44 tok/s (for the first
prompt) and settles at around 26 tok/s on 7900xtx.
2023-09-19 18:17:41 -04:00
..
2023-08-25 23:04:28 -07:00
2023-07-19 23:10:23 +05:30

CodeGen Setup using SHARK-server

Setup Server

  • clone SHARK and setup the venv
  • host the server using python apps/stable_diffusion/web/index.py --api --server_port=<PORT>
  • default server address is http://0.0.0.0:8080

Setup Client

  1. fauxpilot-vscode (VSCode Extension):
  • Code for the extension can be found here
  • PreReq: VSCode extension (will need nodejs and npm to compile and run the extension)
  • Compile and Run the extension on VSCode (press F5 on VSCode), this opens a new VSCode window with the extension running
  • Open VSCode settings, search for fauxpilot in settings and modify server : http://<IP>:<PORT>, Model : codegen , Max Lines : 30
  1. Others (REST API curl, OpenAI Python bindings) as shown here
  • using Github Copilot VSCode extension with SHARK-server needs more work to be functional.