Fixing merge conflict

458d7785 · rprenger · d7bf1ab5 · 9c5a830f · 458d7785 · 458d7785
Commit 458d7785 authored Feb 23, 2022 by rprenger
5 changed files
--- a/tasks/msdp/prompt.py
+++ b/tasks/msdp/prompt.py
--- a/tasks/vision/classification.py
+++ b/tasks/vision/classification.py
--- a/tasks/vision/eval_utils.py
+++ b/tasks/vision/eval_utils.py
--- a/tasks/vision/finetune_utils.py
+++ b/tasks/vision/finetune_utils.py
--- a/tools/run_text_generation_server.py
+++ b/tools/run_text_generation_server.py
@@ -78,4 +78,7 @@ if __name__ == "__main__":
        choice = torch.cuda.LongTensor(1)
        torch.distributed.broadcast(choice, 0)
        if choice[0].item() == 0:
+            try:
                generate_and_post_process(model)
+            except ValueError as ve:
+                pass