From 10077cf4c5d74af2a15116241047b2066df30346 Mon Sep 17 00:00:00 2001 From: decoded Date: Thu, 4 May 2023 09:35:15 -0500 Subject: [PATCH] update --- maple_localized.py | 48 +++++++++++++++++++++++++++------------------- 1 file changed, 28 insertions(+), 20 deletions(-) diff --git a/maple_localized.py b/maple_localized.py index 72eac0b..5024e9d 100644 --- a/maple_localized.py +++ b/maple_localized.py @@ -142,26 +142,34 @@ if missing: ################################################################################################################# for missing_module in missing: - msg(f"{BRK}!") - log("SHELLED",f"INSTALLING MISSING PYTHON MODULE: {str(missing_module).upper()}",4) - msg(f"!{BRK}!") - run_cmd(f'{sys.executable} -um pip install {missing_module}') - if missing_module=='asyncio': - exec(f'import {missing_module}') - elif missing_module=="sklearn": - import sklearn - elif missing_module=="numpy": - import numpy as np - elif missing_module=="requests": - exec(f'import {missing_module}') - elif missing_module=="torch": - exec(f'import {missing_module}') - import torch.nn.functional as F - elif missing_module=="tqdm": - from tqdm import tqdm - elif missing_module=="transformers==2.3.0": - from transformers import GPT2Config, GPT2LMHeadModel, GPT2Tokenizer - + if not missing_module=="transformers==2.3.0": + msg(f"{BRK}!") + log("SHELLED",f"INSTALLING MISSING PYTHON MODULE: {str(missing_module).upper()}",4) + msg(f"!{BRK}!") + run_cmd(f'{sys.executable} -um pip install {missing_module}') + if missing_module=='asyncio': + exec(f'import {missing_module}') + elif missing_module=="sklearn": + import sklearn + elif missing_module=="numpy": + import numpy as np + elif missing_module=="requests": + exec(f'import {missing_module}') + elif missing_module=="torch": + exec(f'import {missing_module}') + import torch.nn.functional as F + elif missing_module=="tqdm": + from tqdm import tqdm + + ################################################################################################################# + ################################################################################################################# + + if "transformers==2.3.0" in missing: + from transformers import GPT2Config, GPT2LMHeadModel, GPT2Tokenizer + +##################################################################################################################### +##################################################################################################################### + if not missing: import asyncio import sklearn