add torch2.3

92ec5d8e · zhuwenwen · 0afe13fb · 92ec5d8e · 92ec5d8e
Commit 92ec5d8e authored Dec 25, 2024 by zhuwenwen
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 6 deletions

README.md README.md +2 -2

setup.py setup.py +4 -4

No files found.
--- a/README.md
+++ b/README.md
@@ -16,10 +16,10 @@ vLLM是一个快速且易于使用的LLM推理和服务库,使用PageAttention
 | ChatGLMModel          | glm-4v-9b,chatglm3,chatglm2                                 | Yes | No  | Yes |
 | DeepseekV2ForCausalLM | DeepSeek-V2                                                 | Yes | No  | -   |
 | BaiChuanForCausalLM   | Baichuan2,Baichuan                                          | Yes | Yes | -   |
-| BloomForCausalLM      | BLOOM                                                       | Yes | No  | -   |
+| BloomForCausalLM      | BLOOM                                                       | Yes | No  | Yes |
 | InternLMForCausalLM   | InternLM                                                    | Yes | No  | -   |
 | InternLM2ForCausalLM  | InternLM2                                                   | Yes | No  | -   |
-| FalconForCausalLM     | falcon                                                      | Yes | No  | -   |
+| FalconForCausalLM     | falcon                                                      | Yes | No  | Yes |
 | TeleChat12BForCausalLM (#TelechatForCausalLM) | TeleChat-12B                        | Yes | No  | -   |
 | MiniCPMForCausalLM    | MiniCPM                                                     | Yes | No  | -   |
 | MiniCPM3ForCausalLM   | MiniCPM3                                                    | Yes | No  | -   |

--- a/setup.py
+++ b/setup.py
@@ -375,13 +375,13 @@ def get_version_add(sha: Optional[str] = None) -> str:
        if sha != 'Unknown':
            if sha is None:
                sha = get_sha(vllm_root)
-            # if (major, minor) == ('2', '3'):
-            #     version = 'das.opt1.' + sha[:7]
+            if (major, minor) == ('2', '3'):
+                version = 'das.opt1.' + sha[:7]
            if (major, minor) == ('2', '4'):
                version = 'das.opt2.' + sha[:7]
    else:
-        # if (major, minor) == ('2', '3'):
-        #     version = 'das.opt1'
+        if (major, minor) == ('2', '3'):
+            version = 'das.opt1'
        if (major, minor) == ('2', '4'):
            version = 'das.opt2'