From 1add0ea51813b541ed2d33169284832aa75fd4c6 Mon Sep 17 00:00:00 2001
From: rombodawg <106640737+rombodawg@users.noreply.github.com>
Date: Tue, 27 Feb 2024 17:41:13 -0500
Subject: [PATCH] Update convert-hf-to-gguf.py

Fixed missing import statements for Gemma support
---
 convert-hf-to-gguf.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py
index ae30b2a76..3a99ef1d7 100755
--- a/convert-hf-to-gguf.py
+++ b/convert-hf-to-gguf.py
@@ -8,10 +8,11 @@ import json
 import os
 import re
 import sys
+import transformers
 from enum import IntEnum
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, ContextManager, Iterator, Sequence, cast
-
+from transformers import GemmaModel
 import numpy as np
 import torch