Spaces:

dwb2023
/

hf_extractor

Running on Zero

dwb2023 commited on Jul 10

Commit

4554904

•

1 Parent(s): 36ecb3d

Update model_utils.py

Files changed (1) hide show

model_utils.py CHANGED Viewed

@@ -22,7 +22,7 @@ ARCHITECTURE_MAP = {
 }
-@spaces.GPU(duration=120)
 def get_model_summary(model_name):
     try:
         config = AutoConfig.from_pretrained(model_name, trust_remote_code=True)

 }
+@spaces.GPU
 def get_model_summary(model_name):
     try:
         config = AutoConfig.from_pretrained(model_name, trust_remote_code=True)