Pass only the HF_TOKEN to the subprocess, which might be required for the script to download gated models
Browse files
app.py
CHANGED
|
@@ -57,12 +57,6 @@ class Config:
|
|
| 57 |
is_using_user_token=bool(user_token),
|
| 58 |
)
|
| 59 |
|
| 60 |
-
def get_env_vars(self) -> Dict[str, str]:
|
| 61 |
-
"""Get environment variables with HF_TOKEN set appropriately."""
|
| 62 |
-
env_vars = os.environ.copy()
|
| 63 |
-
env_vars["HF_TOKEN"] = self.hf_token
|
| 64 |
-
return env_vars
|
| 65 |
-
|
| 66 |
|
| 67 |
class ModelConverter:
|
| 68 |
"""Handles model conversion and upload operations."""
|
|
@@ -128,7 +122,9 @@ class ModelConverter:
|
|
| 128 |
cwd=self.config.repo_path,
|
| 129 |
capture_output=True,
|
| 130 |
text=True,
|
| 131 |
-
env=
|
|
|
|
|
|
|
| 132 |
)
|
| 133 |
|
| 134 |
def convert_model(
|
|
|
|
| 57 |
is_using_user_token=bool(user_token),
|
| 58 |
)
|
| 59 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 60 |
|
| 61 |
class ModelConverter:
|
| 62 |
"""Handles model conversion and upload operations."""
|
|
|
|
| 122 |
cwd=self.config.repo_path,
|
| 123 |
capture_output=True,
|
| 124 |
text=True,
|
| 125 |
+
env={
|
| 126 |
+
"HF_TOKEN": self.config.hf_token,
|
| 127 |
+
},
|
| 128 |
)
|
| 129 |
|
| 130 |
def convert_model(
|