Spaces:

Chengxb888
/

test

Sleeping

Chengxb888 commited on Jul 15, 2024

Commit

744ab35

verified ·

1 Parent(s): 3cc14e8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ def say_hello(msg: str):
     print("model")
     torch.random.manual_seed(0)
     model = AutoModelForCausalLM.from_pretrained(
-        "microsoft/Phi-3-mini-128k-instruct",
         device_map="auto",
         torch_dtype="auto",
         trust_remote_code=True,

     print("model")
     torch.random.manual_seed(0)
     model = AutoModelForCausalLM.from_pretrained(
+        "microsoft/Phi-3-mini-4k-instruct",
         device_map="auto",
         torch_dtype="auto",
         trust_remote_code=True,