Glm4vMoeForConditionalGeneration change to Glm4vForConditionalGeneration
Browse files
README.md
CHANGED
|
@@ -77,7 +77,7 @@ pip install transformers>=5.0.0rc0
|
|
| 77 |
### Quick Start with Transformers
|
| 78 |
|
| 79 |
```python
|
| 80 |
-
from transformers import AutoProcessor,
|
| 81 |
import torch
|
| 82 |
|
| 83 |
MODEL_PATH = "zai-org/GLM-4.6V-Flash"
|
|
@@ -97,7 +97,7 @@ messages = [
|
|
| 97 |
}
|
| 98 |
]
|
| 99 |
processor = AutoProcessor.from_pretrained(MODEL_PATH)
|
| 100 |
-
model =
|
| 101 |
pretrained_model_name_or_path=MODEL_PATH,
|
| 102 |
torch_dtype="auto",
|
| 103 |
device_map="auto",
|
|
|
|
| 77 |
### Quick Start with Transformers
|
| 78 |
|
| 79 |
```python
|
| 80 |
+
from transformers import AutoProcessor, Glm4vForConditionalGeneration
|
| 81 |
import torch
|
| 82 |
|
| 83 |
MODEL_PATH = "zai-org/GLM-4.6V-Flash"
|
|
|
|
| 97 |
}
|
| 98 |
]
|
| 99 |
processor = AutoProcessor.from_pretrained(MODEL_PATH)
|
| 100 |
+
model = Glm4vForConditionalGeneration.from_pretrained(
|
| 101 |
pretrained_model_name_or_path=MODEL_PATH,
|
| 102 |
torch_dtype="auto",
|
| 103 |
device_map="auto",
|