Spaces:

dwb2023
/

hf_extractor

Running on Zero

dwb2023 commited on Jun 15

Commit

f480be7

•

1 Parent(s): 2b69115

Update model_utils.py

Files changed (1) hide show

model_utils.py CHANGED Viewed

@@ -24,7 +24,7 @@ ARCHITECTURE_MAP = {
 @spaces.GPU
 def get_model_summary(model_name):
     try:
-        config = AutoConfig.from_pretrained(model_name)
         architecture = config.architectures[0]
         quantization_config = getattr(config, 'quantization_config', None)

 @spaces.GPU
 def get_model_summary(model_name):
     try:
+        config = AutoConfig.from_pretrained(model_name, trust_remote_code=True)
         architecture = config.architectures[0]
         quantization_config = getattr(config, 'quantization_config', None)