Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -65,7 +65,9 @@ def is_hub_repo_like(s: str) -> bool:
|
|
| 65 |
|
| 66 |
def download_from_hf(repo_id: str, filename: str, token: Optional[str] = None) -> str:
|
| 67 |
token = token or os.environ.get("HF_TOKEN")
|
| 68 |
-
return hf_hub_download(repo_id=repo_id, filename=filename,
|
|
|
|
|
|
|
| 69 |
|
| 70 |
def find_target_modules(model, candidates=("q_proj", "k_proj", "v_proj", "o_proj", "to_q", "to_k", "to_v", "proj_out", "to_out")):
|
| 71 |
names = [n for n, _ in model.named_modules()]
|
|
|
|
| 65 |
|
| 66 |
def download_from_hf(repo_id: str, filename: str, token: Optional[str] = None) -> str:
|
| 67 |
token = token or os.environ.get("HF_TOKEN")
|
| 68 |
+
return hf_hub_download(repo_id=repo_id, filename=filename, repo_type="dataset)
|
| 69 |
+
|
| 70 |
+
|
| 71 |
|
| 72 |
def find_target_modules(model, candidates=("q_proj", "k_proj", "v_proj", "o_proj", "to_q", "to_k", "to_v", "proj_out", "to_out")):
|
| 73 |
names = [n for n, _ in model.named_modules()]
|