rahul7star commited on
Commit
a2dc5f7
·
verified ·
1 Parent(s): f61a819

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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, use_auth_token=token)
 
 
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()]