Skip to content

Commit

Permalink
fix flake8 errors
Browse files Browse the repository at this point in the history
  • Loading branch information
danbev committed Nov 4, 2024
1 parent ba091b3 commit 7caf944
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions convert_hf_to_gguf.py
Original file line number Diff line number Diff line change
Expand Up @@ -1662,6 +1662,7 @@ def prepare_tensors(self):
if len(experts) > 0:
raise ValueError(f"Unprocessed experts: {experts}")


@Model.register("MllamaForConditionalGeneration")
class MLlamaModel(Model):
model_arch = gguf.MODEL_ARCH.MLLAMA
Expand All @@ -1673,10 +1674,11 @@ def find_hparam(self, keys: Iterable[str], optional: bool = False) -> Any:
return super().find_hparam(keys, optional)

def set_vocab(self):
logger.info(f"[danbev] set_vocab...")
logger.info("[danbev] set_vocab...")

def set_gguf_parameters(self):
logger.info(f"[danbev] set_gguf_parameters...")
logger.info("[danbev] set_gguf_parameters...")


@Model.register("BitnetForCausalLM")
class BitnetModel(Model):
Expand Down

0 comments on commit 7caf944

Please sign in to comment.