Commit d5195462 authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

add warning re: _get_backend behavior

parent dffce65e
...@@ -444,6 +444,8 @@ class HFLM(TemplateLM): ...@@ -444,6 +444,8 @@ class HFLM(TemplateLM):
# escape hatch: if we're using a subclass that shouldn't follow # escape hatch: if we're using a subclass that shouldn't follow
# the default _get_backend logic, # the default _get_backend logic,
# then skip over the method. # then skip over the method.
# TODO: this seems very much undesirable in some cases--our code in HFLM
# references AutoModelForCausalLM at times to check for equality
if self.AUTO_MODEL_CLASS is not None: if self.AUTO_MODEL_CLASS is not None:
return return
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment