mlabonne commited on
Commit
8f85465
·
verified ·
1 Parent(s): 7bf890c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -400,7 +400,7 @@ def merge_loop():
400
  df = make_df("open-llm-leaderboard.csv", N_ROWS)
401
 
402
  # Sample two models
403
- print(f"Total size: {human_readable_size(get_size('.'))}")
404
  sample = df.sample(n=2)
405
  models = [sample.iloc[i] for i in range(2)]
406
 
@@ -416,30 +416,30 @@ def merge_loop():
416
  # Merge configs
417
  yaml_config = create_config(models)
418
  print(f"YAML config:{yaml_config}")
419
- print(f"Total size: {human_readable_size(get_size('.'))}")
420
 
421
  # Merge models
422
  merge_models()
423
  print("Model merged!")
424
- print(f"Total size: {human_readable_size(get_size('.'))}")
425
 
426
  # Create model card
427
  print("Create model card")
428
  create_model_card(yaml_config, model_name, USERNAME, license)
429
- print(f"Total size: {human_readable_size(get_size('.'))}")
430
 
431
  # Upload model
432
  print("Upload model")
433
  upload_model(api, USERNAME, model_name)
434
- print(f"Total size: {human_readable_size(get_size('.'))}")
435
 
436
  # Clear data
437
  clear_data()
438
- print(f"Total size: {human_readable_size(get_size('.'))}")
439
 
440
  # Evaluate model on Runpod
441
  create_pod(model_name, USERNAME)
442
- print(f"Total size: {human_readable_size(get_size('.'))}")
443
 
444
 
445
  # Install scrape-open-llm-leaderboard and mergekit
 
400
  df = make_df("open-llm-leaderboard.csv", N_ROWS)
401
 
402
  # Sample two models
403
+ print(f"Total size: {human_readable_size(get_size('/'))}")
404
  sample = df.sample(n=2)
405
  models = [sample.iloc[i] for i in range(2)]
406
 
 
416
  # Merge configs
417
  yaml_config = create_config(models)
418
  print(f"YAML config:{yaml_config}")
419
+ print(f"Total size: {human_readable_size(get_size('/'))}")
420
 
421
  # Merge models
422
  merge_models()
423
  print("Model merged!")
424
+ print(f"Total size: {human_readable_size(get_size('/'))}")
425
 
426
  # Create model card
427
  print("Create model card")
428
  create_model_card(yaml_config, model_name, USERNAME, license)
429
+ print(f"Total size: {human_readable_size(get_size('/'))}")
430
 
431
  # Upload model
432
  print("Upload model")
433
  upload_model(api, USERNAME, model_name)
434
+ print(f"Total size: {human_readable_size(get_size('/'))}")
435
 
436
  # Clear data
437
  clear_data()
438
+ print(f"Total size: {human_readable_size(get_size('/'))}")
439
 
440
  # Evaluate model on Runpod
441
  create_pod(model_name, USERNAME)
442
+ print(f"Total size: {human_readable_size(get_size('/'))}")
443
 
444
 
445
  # Install scrape-open-llm-leaderboard and mergekit