aws_test

Sleeping

Hjgugugjhuhjggg commited on Dec 27, 2024

Commit

3ed39a1

verified ·

1 Parent(s): 5d4a408

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -95,7 +95,9 @@ class S3ModelLoader:
             return model, tokenizer
         except EnvironmentError:
             try:
-                config = AutoConfig.from_pretrained(model_name)
                 tokenizer = AutoTokenizer.from_pretrained(
                     model_name, config=config, token=HUGGINGFACE_HUB_TOKEN
                 )

             return model, tokenizer
         except EnvironmentError:
             try:
+                config = AutoConfig.from_pretrained(
+                    model_name, token=HUGGINGFACE_HUB_TOKEN
+                )
                 tokenizer = AutoTokenizer.from_pretrained(
                     model_name, config=config, token=HUGGINGFACE_HUB_TOKEN
                 )