lora.rs 913 Bytes
Newer Older
1
// SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2
3
4
5
6
7
// SPDX-License-Identifier: Apache-2.0

//! LoRA downloading and caching infrastructure
//!
//! This module provides a minimal, extensible interface for downloading LoRA adapters
//! from various sources (local filesystem, S3, etc.) with automatic caching.
8
9
//! It also provides routing and lora allocation algorithms for distributing LoRA adapters
//! across workers in a cluster.
10
11
12

mod cache;
mod downloader;
13
pub mod load_estimator;
14
pub mod routing;
15
16
17
18
mod source;

pub use cache::LoRACache;
pub use downloader::LoRADownloader;
19
pub use load_estimator::{LoadEstimator, LoadEstimatorConfig, LoadSample};
20
21
22
23
pub use routing::{
    AllocationAlgorithmType, LoraAllocator, LoraReplicaConfig, LoraRoutingTable, RendezvousHasher,
    create_lora_allocator,
};
24
pub use source::{LoRASource, LocalLoRASource, S3LoRASource};