user55442 commited on
Commit
a9187dc
·
verified ·
1 Parent(s): ad0d373

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -40,10 +40,14 @@ base_model = "unsloth/Qwen3-VL-2B-Instruct"
40
  adapter = "user55442/Vilnius-Bus-Stop-LLM"
41
 
42
  model = Qwen2VLForConditionalGeneration.from_pretrained(
43
- base_model, torch_dtype=torch.bfloat16, device_map="auto"
 
 
 
44
  )
45
  model = PeftModel.from_pretrained(model, adapter)
46
  processor = AutoProcessor.from_pretrained(base_model)
 
47
  ```
48
 
49
  ## Training Details
 
40
  adapter = "user55442/Vilnius-Bus-Stop-LLM"
41
 
42
  model = Qwen2VLForConditionalGeneration.from_pretrained(
43
+ base_model,
44
+ torch_dtype=torch.bfloat16,
45
+ device_map="auto",
46
+ ignore_mismatched_sizes=True
47
  )
48
  model = PeftModel.from_pretrained(model, adapter)
49
  processor = AutoProcessor.from_pretrained(base_model)
50
+
51
  ```
52
 
53
  ## Training Details