Spaces:

RamAnanth1
/

T2I-Adapter

Runtime error

RamAnanth1 commited on Feb 18, 2023

Commit

28425ed

1 Parent(s): 5914cbe

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -53,7 +53,7 @@ class Model:
                  use_lightweight: bool = True):
         self.device = torch.device(
             'cuda:0' if torch.cuda.is_available() else 'cpu')
-        self.model_dir = model_dir
         self.download_models()

                  use_lightweight: bool = True):
         self.device = torch.device(
             'cuda:0' if torch.cuda.is_available() else 'cpu')
+        self.model_dir = pathlib.Path(model_dir)
         self.download_models()