TEN-framework
/

TEN_Turn_Detection

Text Generation

natural language understanding

Model card Files Files and versions

zrj619 commited on May 15, 2025

Commit

f6166d7

·

verified ·

1 Parent(s): a102eb2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -97,7 +97,7 @@ import torch
 # Load model and tokenizer
 model_id = 'TEN-framework/TEN_Turn_Detection'
-model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True, low_cpu_mem_usage=True, torch_dtype=torch.bfloat16)
 tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
 # Move model to GPU

 # Load model and tokenizer
 model_id = 'TEN-framework/TEN_Turn_Detection'
+model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True, torch_dtype=torch.bfloat16)
 tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
 # Move model to GPU