snowclipsed
commited on
Commit
·
fa2446a
1
Parent(s):
7090fbf
add back passing dtype in build text model
Browse files- moondream.py +1 -1
moondream.py
CHANGED
|
@@ -75,7 +75,7 @@ class MoondreamModel(nn.Module):
|
|
| 75 |
"vikhyatk/moondream2", revision="2025-01-09"
|
| 76 |
)
|
| 77 |
self.vision = build_vision_model(config.vision, dtype)
|
| 78 |
-
self.text = build_text_model(config.text)
|
| 79 |
|
| 80 |
# Region Model
|
| 81 |
linear_cls = (
|
|
|
|
| 75 |
"vikhyatk/moondream2", revision="2025-01-09"
|
| 76 |
)
|
| 77 |
self.vision = build_vision_model(config.vision, dtype)
|
| 78 |
+
self.text = build_text_model(config.text, dtype)
|
| 79 |
|
| 80 |
# Region Model
|
| 81 |
linear_cls = (
|