Spaces:

Didier
/

Text_translation

Sleeping

Didier commited on Sep 19, 2024

Commit

eec4fa3

verified ·

1 Parent(s): 6b417b7

Update model_translation.py

Files changed (1) hide show

model_translation.py CHANGED Viewed

@@ -2,7 +2,7 @@
 File: model_translation.py
 Description:
-   Loading models for text translations (EN->FR, FR->EN)
 Author: Didier Guillevic
 Date: 2024-03-16
@@ -70,7 +70,7 @@ model_m2m100 = M2M100ForConditionalGeneration.from_pretrained(
     device_map="auto",
     torch_dtype=torch.float16,
     low_cpu_mem_usage=True,
-    load_in_8_bit=True
 )
 #

 File: model_translation.py
 Description:
+   Loading models for text translations
 Author: Didier Guillevic
 Date: 2024-03-16
     device_map="auto",
     torch_dtype=torch.float16,
     low_cpu_mem_usage=True,
+    load_in_8bit=True
 )
 #