Enkhai
/

gpt-j-6b-8bit-lora

Text Generation

Model card Files Files and versions Community

Enkhai commited on Mar 20, 2023

Commit

16cec6d

•

1 Parent(s): 257f9e3

Update gptj.py

Files changed (1) hide show

gptj.py +1 -1

gptj.py CHANGED Viewed

@@ -75,6 +75,6 @@ class GPTJLoraForCausalLM(transformers.models.gptj.modeling_gptj.GPTJForCausalLM
 transformers.models.gptj.modeling_gptj.GPTJBlock = GPTJLoraBlock  # monkey-patch GPT-J
-transformers.AutoConfig.register("gptj", GPTJLoraConfig)
 transformers.AutoModelForCausalLM.register(GPTJLoraConfig, GPTJLoraForCausalLM)

 transformers.models.gptj.modeling_gptj.GPTJBlock = GPTJLoraBlock  # monkey-patch GPT-J
+transformers.AutoConfig.register("gptj-lora", GPTJLoraConfig)
 transformers.AutoModelForCausalLM.register(GPTJLoraConfig, GPTJLoraForCausalLM)