bebechien commited on
Commit
e7dbe77
·
verified ·
1 Parent(s): 4a7bf0e

Fix repo_id

Browse files
Files changed (1) hide show
  1. engine.py +2 -2
engine.py CHANGED
@@ -426,7 +426,7 @@ class FunctionGemmaEngine:
426
  print(f"Uploading to {repo_url.repo_id}...")
427
  api.upload_folder(
428
  folder_path=str(self.output_dir),
429
- repo_id=repo_name,
430
  repo_type="model"
431
  )
432
 
@@ -436,7 +436,7 @@ class FunctionGemmaEngine:
436
  )
437
  tags = info.card_data.tags
438
  tags.append("functiongemma-tuning-lab")
439
- metadata_update(repo_id, {"tags": tags}, overwrite=True, token=oauth_token)
440
 
441
  return f"✅ Success! Model uploaded to: {repo_url}"
442
  except Exception as e:
 
426
  print(f"Uploading to {repo_url.repo_id}...")
427
  api.upload_folder(
428
  folder_path=str(self.output_dir),
429
+ repo_id=repo_url.repo_id,
430
  repo_type="model"
431
  )
432
 
 
436
  )
437
  tags = info.card_data.tags
438
  tags.append("functiongemma-tuning-lab")
439
+ metadata_update(repo_url.repo_id, {"tags": tags}, overwrite=True, token=oauth_token)
440
 
441
  return f"✅ Success! Model uploaded to: {repo_url}"
442
  except Exception as e: