PrashantGoyal commited on
Commit
1f43103
·
1 Parent(s): 5c77158

model issue solving

Browse files
Files changed (2) hide show
  1. src/evaluation.py +1 -0
  2. src/training.py +1 -0
src/evaluation.py CHANGED
@@ -16,6 +16,7 @@ MODEL_ID = "PrashantGoyal/findr-clip-ft"
16
 
17
  model_path = hf_hub_download(
18
  repo_id=MODEL_ID,
 
19
  filename="clip/best.pt",
20
  token=os.getenv("HF_TOKEN")
21
  )
 
16
 
17
  model_path = hf_hub_download(
18
  repo_id=MODEL_ID,
19
+ force_download=True,
20
  filename="clip/best.pt",
21
  token=os.getenv("HF_TOKEN")
22
  )
src/training.py CHANGED
@@ -23,6 +23,7 @@ MODEL_ID = "PrashantGoyal/findr-clip-ft"
23
 
24
  model_path = hf_hub_download(
25
  repo_id=MODEL_ID,
 
26
  filename="clip/best.pt",
27
  token=os.getenv("HF_TOKEN")
28
  )
 
23
 
24
  model_path = hf_hub_download(
25
  repo_id=MODEL_ID,
26
+ force_download=True,
27
  filename="clip/best.pt",
28
  token=os.getenv("HF_TOKEN")
29
  )