parent
e3fb5a5de4
commit
d8539ffc9e
@ -0,0 +1,4 @@
|
|||||||
|
{
|
||||||
|
"display_message": "dismiss",
|
||||||
|
"version": "0.1.0"
|
||||||
|
}
|
@ -0,0 +1,48 @@
|
|||||||
|
def splash():
|
||||||
|
import json
|
||||||
|
|
||||||
|
with open('config.json', 'r') as file:
|
||||||
|
config = json.load(file)
|
||||||
|
|
||||||
|
if config["display_message"]:
|
||||||
|
config_display_message = config["display_message"]
|
||||||
|
else:
|
||||||
|
config["display_message"] = ""
|
||||||
|
config_display_message = config["display_message"]
|
||||||
|
|
||||||
|
if config["version"]:
|
||||||
|
config_version = config["version"]
|
||||||
|
else:
|
||||||
|
config["version"] = ""
|
||||||
|
config_version = config["version"]
|
||||||
|
|
||||||
|
env_file_empty = config_display_message == ""
|
||||||
|
|
||||||
|
if env_file_empty or config_display_message == "keep":
|
||||||
|
print("""
|
||||||
|
|
||||||
|
Warning: Some packages may not have been installed correctly. to ensure all packages are installed, run:
|
||||||
|
pip install python-dotenv
|
||||||
|
pip install termcolor
|
||||||
|
pip install setuptools
|
||||||
|
Pytorch will install with a default cpu config that may be different then the one for your machine. if this causes errors, run:
|
||||||
|
pip uninstall pytorch
|
||||||
|
pip install pytorch
|
||||||
|
This message will only display the first time swarms is imported by default.
|
||||||
|
Hit any key to continue, type "keep" | "yes" | "y" to keep showing this message, type "dismiss" to not show this message again
|
||||||
|
""")
|
||||||
|
|
||||||
|
|
||||||
|
continue_or_not = input("keep showing message? ")
|
||||||
|
|
||||||
|
if (env_file_empty and continue_or_not != "keep") or (config_display_message == "keep" and continue_or_not == "dismiss"):
|
||||||
|
config["display_message"] = "dismiss"
|
||||||
|
elif continue_or_not == "keep" or continue_or_not == None:
|
||||||
|
config["display_message"] = "keep"
|
||||||
|
|
||||||
|
with open('config.json', 'w') as file:
|
||||||
|
json.dump(config, file, indent=4)
|
||||||
|
|
||||||
|
print(f"version: {config_version}")
|
||||||
|
print("some packages may not be installed correctly. see __init__.py for more details. ")
|
||||||
|
print("importing swarms... ")
|
Loading…
Reference in new issue