Missed 1 file.

This commit is contained in:
Michael Chung committed 2023-12-21 00:42:22 -05:00
1 parent 1bddb21e9f
commit eca8b01f14
1 file changed
+48
@@ -0,0 +1,48 @@
use super::file_descriptor::FileDescriptor;
use crate::remote_file_manager::weights_descriptor::{WeightsLoRADescriptor, WeightsSD15Descriptor, WeightsSDXLDescriptor, self};
pub struct RemoteCloudBucketDetails {
object_hash: String,
prefix: String,
suffix: String,
}
impl RemoteCloudBucketDetails {
pub fn new(object_hash: String, prefix: String, suffix: String) -> Self {
Self {
object_hash: object_hash,
prefix: prefix,
suffix: suffix
}
}
pub fn get_object_hash(&self) -> &str {
&self.object_hash
}
pub fn get_prefix(&self) -> &str {
&self.prefix
}
pub fn get_suffix(&self) -> &str {
&self.suffix
}
pub fn file_descriptor_from_bucket_details(&self) -> Box<dyn FileDescriptor> {
match self.prefix.as_str() {
"loRA" => Box::new(weights_descriptor::WeightsLoRADescriptor {}),
"SD15" => Box::new(weights_descriptor::WeightsSD15Descriptor {}),
"SDXL" => Box::new(weights_descriptor::WeightsSDXLDescriptor {}),
_ => panic!("Unknown prefix: {}", self.prefix)
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
pub fn test() {}
}