#57 Added single file option

README.md

@@ -20,5 +20,5 @@ We welcome contributions from everyone. There are several ways to contribute, an
 
 ## License
 
-All code is licensed under the TBD. See [LICENSE](./LICENSE.txt) file for 
+All code is licensed under the TBD. See [LICENSE](./LICENSE) file for 
 details.

sdk/models/model.py

@@ -11,23 +11,25 @@ class Model:
     model_name: str
     model_path: str
     device: Union[str, Devices]
-
     loaded: bool
+    single_file: bool
 
     def __init__(self, model_name, model_path: str,
-                 device: Union[str, Devices]):
+                 device: Union[str, Devices], single_file: bool = False):
         """
         Initializes the model with the given name
 
         Args:
             model_name (str): The name of the model
             model_path (str): The path of the model
             device (Union[str, Devices]): Which device the model must be on
+            :param single_file: Whether model is single file or not
         """
         self.model_name = model_name
         self.model_path = model_path
         self.device = device
         self.loaded = False
+        self.single_file = single_file
 
     @abstractmethod
     def load_model(self) -> bool:

sdk/models/model_diffusers.py

@@ -41,6 +41,14 @@ def create_pipeline(self, **kwargs) -> None:
         if self.loaded:
             return
 
+        # Single file loads a single ckpt/safetensors file
+        if self.single_file:
+            self.pipeline = self.model_class.from_single_file(
+                self.model_path,
+                **kwargs
+            )
+            return
+
         self.pipeline = self.model_class.from_pretrained(
             self.model_path,
             **kwargs