Fetch files metadata directly from model_info call

#1
by Wauplin HF staff - opened
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from huggingface_hub import hf_hub_url, get_hf_file_metadata, model_info
2
  import gradio as gr
3
 
4
 
@@ -22,7 +22,7 @@ def get_component_wise_memory(pipeline_id, token=None, variant=None, revision=No
22
  print(pipeline_id, variant, revision, extension)
23
  component_wise_memory = {}
24
 
25
- files_in_repo = model_info(pipeline_id, revision=revision, token=token).siblings
26
 
27
  for current_file in files_in_repo:
28
  if all(
@@ -44,10 +44,7 @@ def get_component_wise_memory(pipeline_id, token=None, variant=None, revision=No
44
  filename = current_file.rfilename
45
 
46
  if filename is not None:
47
- hub_url = hf_hub_url(repo_id=pipeline_id, filename=filename)
48
- file_metadata = get_hf_file_metadata(hub_url)
49
-
50
- component_wise_memory[component] = bytes_to_giga_bytes(file_metadata.size)
51
 
52
  return component_wise_memory
53
 
 
1
+ from huggingface_hub import model_info
2
  import gradio as gr
3
 
4
 
 
22
  print(pipeline_id, variant, revision, extension)
23
  component_wise_memory = {}
24
 
25
+ files_in_repo = model_info(pipeline_id, revision=revision, token=token, files_metadata=True).siblings
26
 
27
  for current_file in files_in_repo:
28
  if all(
 
44
  filename = current_file.rfilename
45
 
46
  if filename is not None:
47
+ component_wise_memory[component] = bytes_to_giga_bytes(current_file.size)
 
 
 
48
 
49
  return component_wise_memory
50