Hugging Face's logo

Spaces:
Didier
/
Text_translation
Runtime error

App Files Community
Didier commited on
Commit
fcf02be
·
1 Parent(s): efcd81a

Forgot comma

Browse files
Files changed (1) hide show
  1. model_translation.py +1 -1
model_translation.py CHANGED
@@ -67,6 +67,6 @@ tokenizer_multilingual = AutoTokenizer.from_pretrained(model_MADLAD_name, use_fa
67
  model_multilingual = AutoModelForSeq2SeqLM.from_pretrained(
68
  model_MADLAD_name,
69
  device_map="auto",
70
- torch_dtype=torch.float16
71
  low_cpu_mem_usage=True)
72
 
 
67
  model_multilingual = AutoModelForSeq2SeqLM.from_pretrained(
68
  model_MADLAD_name,
69
  device_map="auto",
70
+ torch_dtype=torch.float16,
71
  low_cpu_mem_usage=True)
72