Hugging Face's logo

Spaces:
dwb2023
/
hf_extractor
Runtime error

App Files Community
dwb2023 commited on
Commit
4554904
·
verified ·
1 Parent(s): 36ecb3d

Update model_utils.py

Browse files
Files changed (1) hide show
  1. model_utils.py +1 -1
model_utils.py CHANGED
@@ -22,7 +22,7 @@ ARCHITECTURE_MAP = {
22
  }
23
 
24
 
25
- @spaces.GPU(duration=120)
26
  def get_model_summary(model_name):
27
  try:
28
  config = AutoConfig.from_pretrained(model_name, trust_remote_code=True)
 
22
  }
23
 
24
 
25
+ @spaces.GPU
26
  def get_model_summary(model_name):
27
  try:
28
  config = AutoConfig.from_pretrained(model_name, trust_remote_code=True)