ErenalpCet
/

E-Model-V2

Text Generation

Model card Files Files and versions Community

ErenalpCet commited on Mar 30

Commit

706ead2

·

verified ·

1 Parent(s): 6941c3f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -89,7 +89,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda"
 # Load the merged fine-tuned model and tokenizer
-model_dir = "E-Model-V1"
 model = AutoModelForCausalLM.from_pretrained(
     model_dir,
     torch_dtype=torch.float16,  # Use FP16 for memory efficiency

 device = "cuda"
 # Load the merged fine-tuned model and tokenizer
+model_dir = "E-Model-V2"
 model = AutoModelForCausalLM.from_pretrained(
     model_dir,
     torch_dtype=torch.float16,  # Use FP16 for memory efficiency