jordiclive
/

falcon-40b-lora-sft-stage2-1.1k

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

jordiclive commited on Jul 13, 2023

Commit

2532041

•

1 Parent(s): f791de6

Update README.md

Files changed (1) hide show

README.md +0 -1

README.md CHANGED Viewed

@@ -27,7 +27,6 @@ dtype = torch.bfloat16
 tokenizer = AutoTokenizer.from_pretrained(repo_id)
 model = AutoModelForCausalLM.from_pretrained(
     repo_id,
-    subfolder="merged_model",
     torch_dtype=dtype,
     trust_remote_code=True,
 )

 tokenizer = AutoTokenizer.from_pretrained(repo_id)
 model = AutoModelForCausalLM.from_pretrained(
     repo_id,
     torch_dtype=dtype,
     trust_remote_code=True,
 )