junnei commited on
Commit
20b9684
·
verified ·
1 Parent(s): cb48fb8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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" # 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
 
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