Merge pull request #393 from Anhforth/masternew

released 1.7.3

Author: ftgreat

BAAI_Aquila_Model_License.pdf

@@ -11,7 +11,7 @@
 loader = AutoLoader(task_name="text2img", #contrastive learning
                     model_name="AltDiffusion-m18",
                     model_dir="./checkpoints",
-                    use_fp16=False)
+                    fp16=False)
 model = loader.get_model()
 model.eval()
 model.to(device)

examples/AltDiffusion-m18/generate.py

@@ -11,7 +11,7 @@
 loader = AutoLoader(task_name="text2img", #contrastive learning
                     model_name="AltDiffusion-m9",
                     model_dir="./checkpoints",
-                    use_fp16=False)
+                    fp16=False)
 
 model = loader.get_model()
 model.eval()

examples/AltDiffusion/generate.py

@@ -220,8 +220,6 @@ def __init__(self,
             only_download_config=only_download_config,
             device=device,
             **kwargs)
-        if kwargs.get("use_fp16", None):
-            self.model.half()
 
         if model_type == "nlp":
             if brief_model_name in ["galactica",]:

flagai/auto_model/auto_loader.py

@@ -5,7 +5,7 @@
 
 setup(
     name="flagai",
-    version="v1.7.2",
+    version="v1.7.3",
     description="FlagAI aims to help researchers and developers to freely train and test large-scale models for NLP/CV/VL tasks.",
     long_description=open("README.md", encoding="utf-8").read(),
     long_description_content_type="text/markdown",
@@ -36,5 +36,7 @@
         'taming-transformers-rom1504 == 0.0.6',
         'rouge-score == 0.1.2',
         'sacrebleu == 2.3.1',
+        'jsonlines == 3.1.0',
+        'accelerate == 0.19.0'
     ]
 )