Spaces:

wjjessen
/

rasa

Running

wjjessen commited on Nov 28, 2023

Commit

b8f16a6

•

1 Parent(s): d124ecd

update code

Files changed (1) hide show

app.py CHANGED Viewed

@@ -128,7 +128,7 @@ def main():
                     model_max_length=1000,
                     trust_remote_code=True,
                 )
-                if model_source == "Download":
                     base_model = AutoModelForSeq2SeqLM.from_pretrained(
                     checkpoint,
                     torch_dtype=torch.float32,
@@ -145,7 +145,7 @@ def main():
                     model_max_length=1000,
                     #cache_dir="model_cache"
                 )
-                if model_source == "Download":
                     base_model = AutoModelForSeq2SeqLM.from_pretrained(
                     checkpoint,
                     torch_dtype=torch.float32,

                     model_max_length=1000,
                     trust_remote_code=True,
                 )
+                if model_source == "Download model":
                     base_model = AutoModelForSeq2SeqLM.from_pretrained(
                     checkpoint,
                     torch_dtype=torch.float32,
                     model_max_length=1000,
                     #cache_dir="model_cache"
                 )
+                if model_source == "Download model":
                     base_model = AutoModelForSeq2SeqLM.from_pretrained(
                     checkpoint,
                     torch_dtype=torch.float32,