Spaces:

ybelkada
/

detoxified-lms

Runtime error

ybelkada commited on Feb 18, 2023

Commit

be1aefc

•

1 Parent(s): 2660039

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ detoxified_gpt_neo_id = "ybelkada/gpt-neo-125m-detoxified-small-context"
 gpt_neo = AutoModelForCausalLM.from_pretrained(gpt_neo_125_id).to(0)
 detoxified_neo = AutoModelForCausalLM.from_pretrained(detoxified_gpt_neo_id).to(0)
-gpt_neo_1b_id = "EleutherAI/gpt-neo-2.7B"
 detoxified_gpt_neo_1b_id = "ybelkada/gpt-neo-2.7B-detoxified-20shdl"
 gpt_neo_1b = AutoModelForCausalLM.from_pretrained(gpt_neo_1b_id, torch_dtype=torch.bfloat16).to(0)

 gpt_neo = AutoModelForCausalLM.from_pretrained(gpt_neo_125_id).to(0)
 detoxified_neo = AutoModelForCausalLM.from_pretrained(detoxified_gpt_neo_id).to(0)
+gpt_neo_1b_id = "ybelkada/gpt-neo-2.7B-sharded-bf16"
 detoxified_gpt_neo_1b_id = "ybelkada/gpt-neo-2.7B-detoxified-20shdl"
 gpt_neo_1b = AutoModelForCausalLM.from_pretrained(gpt_neo_1b_id, torch_dtype=torch.bfloat16).to(0)