Update README.md
Browse files
README.md
CHANGED
@@ -111,7 +111,7 @@ from transformers import AutoProcessor, AutoModel
|
|
111 |
import torch
|
112 |
|
113 |
model_id = "junnei/gemma-3-4b-it-speech"
|
114 |
-
revision = "v1.0"
|
115 |
|
116 |
model = AutoModel.from_pretrained(
|
117 |
model_id, device_map="auto", revision = revision, trust_remote_code=True
|
|
|
111 |
import torch
|
112 |
|
113 |
model_id = "junnei/gemma-3-4b-it-speech"
|
114 |
+
revision = "main" # or "v1.0". cause model is being updated, main branch could be crashed.
|
115 |
|
116 |
model = AutoModel.from_pretrained(
|
117 |
model_id, device_map="auto", revision = revision, trust_remote_code=True
|