| .. |
|
__init__.py
|
04b53d2db5
chore: add initializer files
|
hace 1 año |
|
cache_engine.py
|
5289c14b24
feat: Asymmetric Tensor Parallel (#594)
|
hace 4 meses |
|
cpu_model_runner.py
|
99680b2d23
feat: soft prompts (#589)
|
hace 4 meses |
|
cpu_worker.py
|
99680b2d23
feat: soft prompts (#589)
|
hace 4 meses |
|
embedding_model_runner.py
|
5289c14b24
feat: Asymmetric Tensor Parallel (#594)
|
hace 4 meses |
|
model_runner.py
|
6671e3a162
feat: add CPU offloading support (#598)
|
hace 4 meses |
|
model_runner_base.py
|
a3e26391e4
chore: add a wrapper for torch.inference_mode decorator
|
hace 4 meses |
|
neuron_model_runner.py
|
4599c98f99
feat: dynamic image size support for VLMs
|
hace 4 meses |
|
neuron_worker.py
|
ae04f57ec1
feat: Pipeline Parallel support (#581)
|
hace 4 meses |
|
openvino_model_runner.py
|
4f7d212b70
feat: remove vision language config
|
hace 4 meses |
|
openvino_worker.py
|
1ff6d4c3d7
feat: support pipeline parallel on indivisible GPU count (#587)
|
hace 4 meses |
|
tpu_model_runner.py
|
f91991f584
fix: f-string fixes
|
hace 4 meses |
|
tpu_worker.py
|
b6c4dfce23
chore: refactor TPU model runner and worker
|
hace 4 meses |
|
worker.py
|
e8af0d4a3b
fix: type annotation in worker
|
hace 4 meses |
|
worker_base.py
|
a3e26391e4
chore: add a wrapper for torch.inference_mode decorator
|
hace 4 meses |
|
xpu_model_runner.py
|
99680b2d23
feat: soft prompts (#589)
|
hace 4 meses |
|
xpu_worker.py
|
99680b2d23
feat: soft prompts (#589)
|
hace 4 meses |
AlpinDale
f91991f584
fix: f-string fixes