Eliasj42 dfdd3b1f78 improved sharded performance and fixed issue with lmhead on rocm (#2008) ...

* improved sharded performance and fixed issue with lmhead on rocm

* mmap shards + disable sharing of device arrays across devices

* fix device_idx for non-layer vmfbs

* fix time calc for sharded

---------

Co-authored-by: Elias Joseph <elias@nod-labs.com>
Co-authored-by: PhaneeshB <b.phaneesh@gmail.com>

2023-12-05 11:53:44 -08:00

..

language_models

improved sharded performance and fixed issue with lmhead on rocm (#2008)

2023-12-05 11:53:44 -08:00

shark_studio

(SHARK Studio) Add Turbine-based llm chatbot. (#1933)

2023-11-14 09:56:28 -06:00

stable_diffusion

Fixes to llama2 cpu compilation and studio UI, schedulers (#2013)

2023-12-05 11:19:19 -05:00

__init__.py

[SD] Reorganize the stable diffusion model. (#806)

2023-01-31 14:42:41 -08:00