david/aphrodite-engine: PygmalionAI's large-scale inference engine pygmalion.chat It is designed to serve as the inference endpoint for the PygmalionAI website, and to allow serving the Pygmalion models to a large number of users with blazing fast speeds (thanks to vLLM's Paged Attention). @ rc_054

AlpinDale a3e26391e4 chore: add a wrapper for torch.inference_mode decorator		4 months ago
..
__init__.py	a3e26391e4 chore: add a wrapper for torch.inference_mode decorator	4 months ago
cuda.py	1a40bf438b fix: incorrect gpu capability when used mixed gpus	4 months ago
interface.py	a3e26391e4 chore: add a wrapper for torch.inference_mode decorator	4 months ago
rocm.py	cda0e93a10 abstract away the platform for device capability	4 months ago
tpu.py	a3e26391e4 chore: add a wrapper for torch.inference_mode decorator	4 months ago