Samsung
/

BigTranslateSlotTranslator

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

pawlowskipawel commited on Apr 11

Commit

2a15446

•

1 Parent(s): 5af5686

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -30,7 +30,7 @@ def translate(input_text, src_lang, trg_lang):
     return tokenizer.decode(generated_tokens, skip_special_tokens=True)[len(prompt):]
-model = AutoModelForCausalLM.from_pretrained("Samsung/BigTranslateSlotTranslator", torch_dtype=torch.float16)
 tokenizer = AutoTokenizer.from_pretrained("Samsung/BigTranslateSlotTranslator")
 translation = translate("set the temperature on my <a>thermostat<a> to <b> 29 degrees <b>", "en", "de")  # translation: stell die temperatur auf meinem <a> thermostat <a> auf <b> 29 grad <b>

     return tokenizer.decode(generated_tokens, skip_special_tokens=True)[len(prompt):]
+model = AutoModelForCausalLM.from_pretrained("Samsung/BigTranslateSlotTranslator")
 tokenizer = AutoTokenizer.from_pretrained("Samsung/BigTranslateSlotTranslator")
 translation = translate("set the temperature on my <a>thermostat<a> to <b> 29 degrees <b>", "en", "de")  # translation: stell die temperatur auf meinem <a> thermostat <a> auf <b> 29 grad <b>