Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -63,7 +63,7 @@ if "client" not in state:
|
|
63 |
api_key=os.getenv("OPENAI_API_KEY"),
|
64 |
base_url=os.getenv("BASE_URL")
|
65 |
)
|
66 |
-
state.model_name = "internlm/internlm2_5-
|
67 |
# model_path = "internlm/internlm2_5-7b-chat"
|
68 |
# state.model = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype=torch.float16, trust_remote_code=True)
|
69 |
# state.tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|
|
|
63 |
api_key=os.getenv("OPENAI_API_KEY"),
|
64 |
base_url=os.getenv("BASE_URL")
|
65 |
)
|
66 |
+
state.model_name = "internlm/internlm2_5-20b-chat"
|
67 |
# model_path = "internlm/internlm2_5-7b-chat"
|
68 |
# state.model = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype=torch.float16, trust_remote_code=True)
|
69 |
# state.tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|