git init before git lfs

gradio_demo/runner.py

@@ -29,14 +29,14 @@ class Runner:
 
     def download_base_model(self, base_model_id: str, token=None) -> str:
         model_dir = self.checkpoint_dir / base_model_id
+        org_name = base_model_id.split('/')[0]
+        org_dir = self.checkpoint_dir / org_name
         if not model_dir.exists():
-            org_name = base_model_id.split('/')[0]
-            org_dir = self.checkpoint_dir / org_name
             org_dir.mkdir(exist_ok=True)
         print(f'https://huggingface.co/{base_model_id}')
         if token == None:
-            subprocess.run(shlex.split(f'git lfs install'),
-                cwd=org_dir)
+            subprocess.run(shlex.split(f'git init'), cwd=org_dir)
+            subprocess.run(shlex.split(f'git lfs install'), cwd=org_dir)
             subprocess.run(shlex.split(
                 f'git lfs clone https://huggingface.co/{base_model_id}'),
                             cwd=org_dir)
@@ -90,6 +90,7 @@ class Runner:
         output_dir.mkdir(parents=True, exist_ok=True)
 
         config = OmegaConf.load('configs/black-swan.yaml')
+        import pdb; pdb.set_trace()
         config.pretrained_model_path = self.download_base_model(model_path, token=input_token)
 
         # we remove null-inversion & use fp16 for fast inference on web demo