mihaimasala commited on
Commit
49cc5bf
·
verified ·
1 Parent(s): 31a98c6

Unify Getting Started snippet (consistency + decode slice fix)

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -269,10 +269,12 @@ than Romanian.
269
  ```python
270
  import torch
271
  from PIL import Image
272
- from transformers import Gemma3ForConditionalGeneration, AutoProcessor
273
 
274
  model = Gemma3ForConditionalGeneration.from_pretrained(
275
- "OpenLLM-Ro/RoGemma3-4B-Instruct", torch_dtype=torch.bfloat16, device_map="auto"
 
 
276
  ).eval()
277
  processor = AutoProcessor.from_pretrained("OpenLLM-Ro/RoGemma3-4B-Instruct")
278
 
 
269
  ```python
270
  import torch
271
  from PIL import Image
272
+ from transformers import AutoProcessor, Gemma3ForConditionalGeneration
273
 
274
  model = Gemma3ForConditionalGeneration.from_pretrained(
275
+ "OpenLLM-Ro/RoGemma3-4B-Instruct",
276
+ torch_dtype=torch.bfloat16,
277
+ device_map="auto",
278
  ).eval()
279
  processor = AutoProcessor.from_pretrained("OpenLLM-Ro/RoGemma3-4B-Instruct")
280