Changed device
Browse files- .idea/.gitignore +3 -0
- .idea/ThetaM2V.iml +12 -0
- .idea/inspectionProfiles/Project_Default.xml +140 -0
- .idea/inspectionProfiles/profiles_settings.xml +6 -0
- .idea/misc.xml +7 -0
- .idea/modules.xml +8 -0
- .idea/vcs.xml +6 -0
- app.py +1 -1
.idea/.gitignore
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Default ignored files
|
| 2 |
+
/shelf/
|
| 3 |
+
/workspace.xml
|
.idea/ThetaM2V.iml
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
<?xml version="1.0" encoding="UTF-8"?>
|
| 2 |
+
<module type="PYTHON_MODULE" version="4">
|
| 3 |
+
<component name="NewModuleRootManager">
|
| 4 |
+
<content url="file://$MODULE_DIR$" />
|
| 5 |
+
<orderEntry type="jdk" jdkName="Testing" jdkType="Python SDK" />
|
| 6 |
+
<orderEntry type="sourceFolder" forTests="false" />
|
| 7 |
+
</component>
|
| 8 |
+
<component name="PyDocumentationSettings">
|
| 9 |
+
<option name="format" value="PLAIN" />
|
| 10 |
+
<option name="myDocStringFormat" value="Plain" />
|
| 11 |
+
</component>
|
| 12 |
+
</module>
|
.idea/inspectionProfiles/Project_Default.xml
ADDED
|
@@ -0,0 +1,140 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
<component name="InspectionProjectProfileManager">
|
| 2 |
+
<profile version="1.0">
|
| 3 |
+
<option name="myName" value="Project Default" />
|
| 4 |
+
<inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
|
| 5 |
+
<option name="ignoredPackages">
|
| 6 |
+
<value>
|
| 7 |
+
<list size="127">
|
| 8 |
+
<item index="0" class="java.lang.String" itemvalue="scipy" />
|
| 9 |
+
<item index="1" class="java.lang.String" itemvalue="torchlibrosa" />
|
| 10 |
+
<item index="2" class="java.lang.String" itemvalue="gradio" />
|
| 11 |
+
<item index="3" class="java.lang.String" itemvalue="diffusers" />
|
| 12 |
+
<item index="4" class="java.lang.String" itemvalue="scikit-learn" />
|
| 13 |
+
<item index="5" class="java.lang.String" itemvalue="torch" />
|
| 14 |
+
<item index="6" class="java.lang.String" itemvalue="numpy" />
|
| 15 |
+
<item index="7" class="java.lang.String" itemvalue="torchvision" />
|
| 16 |
+
<item index="8" class="java.lang.String" itemvalue="omegaconf" />
|
| 17 |
+
<item index="9" class="java.lang.String" itemvalue="tqdm" />
|
| 18 |
+
<item index="10" class="java.lang.String" itemvalue="gradio_client" />
|
| 19 |
+
<item index="11" class="java.lang.String" itemvalue="soundfile" />
|
| 20 |
+
<item index="12" class="java.lang.String" itemvalue="librosa" />
|
| 21 |
+
<item index="13" class="java.lang.String" itemvalue="pillow" />
|
| 22 |
+
<item index="14" class="java.lang.String" itemvalue="accelerate" />
|
| 23 |
+
<item index="15" class="java.lang.String" itemvalue="einops" />
|
| 24 |
+
<item index="16" class="java.lang.String" itemvalue="torchaudio" />
|
| 25 |
+
<item index="17" class="java.lang.String" itemvalue="httpx" />
|
| 26 |
+
<item index="18" class="java.lang.String" itemvalue="google-pasta" />
|
| 27 |
+
<item index="19" class="java.lang.String" itemvalue="numba" />
|
| 28 |
+
<item index="20" class="java.lang.String" itemvalue="tensorflow-estimator" />
|
| 29 |
+
<item index="21" class="java.lang.String" itemvalue="ftfy" />
|
| 30 |
+
<item index="22" class="java.lang.String" itemvalue="starlette" />
|
| 31 |
+
<item index="23" class="java.lang.String" itemvalue="astunparse" />
|
| 32 |
+
<item index="24" class="java.lang.String" itemvalue="multiprocess" />
|
| 33 |
+
<item index="25" class="java.lang.String" itemvalue="pyarrow-hotfix" />
|
| 34 |
+
<item index="26" class="java.lang.String" itemvalue="uvicorn" />
|
| 35 |
+
<item index="27" class="java.lang.String" itemvalue="pywin32" />
|
| 36 |
+
<item index="28" class="java.lang.String" itemvalue="libclang" />
|
| 37 |
+
<item index="29" class="java.lang.String" itemvalue="pydantic" />
|
| 38 |
+
<item index="30" class="java.lang.String" itemvalue="transformers" />
|
| 39 |
+
<item index="31" class="java.lang.String" itemvalue="Werkzeug" />
|
| 40 |
+
<item index="32" class="java.lang.String" itemvalue="streamlit" />
|
| 41 |
+
<item index="33" class="java.lang.String" itemvalue="tensorboard-data-server" />
|
| 42 |
+
<item index="34" class="java.lang.String" itemvalue="audioread" />
|
| 43 |
+
<item index="35" class="java.lang.String" itemvalue="pydub" />
|
| 44 |
+
<item index="36" class="java.lang.String" itemvalue="contourpy" />
|
| 45 |
+
<item index="37" class="java.lang.String" itemvalue="flatbuffers" />
|
| 46 |
+
<item index="38" class="java.lang.String" itemvalue="regex" />
|
| 47 |
+
<item index="39" class="java.lang.String" itemvalue="pydantic_core" />
|
| 48 |
+
<item index="40" class="java.lang.String" itemvalue="tensorboard" />
|
| 49 |
+
<item index="41" class="java.lang.String" itemvalue="optree" />
|
| 50 |
+
<item index="42" class="java.lang.String" itemvalue="matplotlib" />
|
| 51 |
+
<item index="43" class="java.lang.String" itemvalue="msgpack" />
|
| 52 |
+
<item index="44" class="java.lang.String" itemvalue="httpcore" />
|
| 53 |
+
<item index="45" class="java.lang.String" itemvalue="rsa" />
|
| 54 |
+
<item index="46" class="java.lang.String" itemvalue="datasets" />
|
| 55 |
+
<item index="47" class="java.lang.String" itemvalue="pyasn1" />
|
| 56 |
+
<item index="48" class="java.lang.String" itemvalue="tensorflow" />
|
| 57 |
+
<item index="49" class="java.lang.String" itemvalue="zipp" />
|
| 58 |
+
<item index="50" class="java.lang.String" itemvalue="ml-dtypes" />
|
| 59 |
+
<item index="51" class="java.lang.String" itemvalue="websockets" />
|
| 60 |
+
<item index="52" class="java.lang.String" itemvalue="annotated-types" />
|
| 61 |
+
<item index="53" class="java.lang.String" itemvalue="importlib_metadata" />
|
| 62 |
+
<item index="54" class="java.lang.String" itemvalue="lazy_loader" />
|
| 63 |
+
<item index="55" class="java.lang.String" itemvalue="watchfiles" />
|
| 64 |
+
<item index="56" class="java.lang.String" itemvalue="soxr" />
|
| 65 |
+
<item index="57" class="java.lang.String" itemvalue="google-auth-oauthlib" />
|
| 66 |
+
<item index="58" class="java.lang.String" itemvalue="pyasn1_modules" />
|
| 67 |
+
<item index="59" class="java.lang.String" itemvalue="aiofiles" />
|
| 68 |
+
<item index="60" class="java.lang.String" itemvalue="python-multipart" />
|
| 69 |
+
<item index="61" class="java.lang.String" itemvalue="tensorflow-io-gcs-filesystem" />
|
| 70 |
+
<item index="62" class="java.lang.String" itemvalue="termcolor" />
|
| 71 |
+
<item index="63" class="java.lang.String" itemvalue="namex" />
|
| 72 |
+
<item index="64" class="java.lang.String" itemvalue="ruff" />
|
| 73 |
+
<item index="65" class="java.lang.String" itemvalue="email_validator" />
|
| 74 |
+
<item index="66" class="java.lang.String" itemvalue="multidict" />
|
| 75 |
+
<item index="67" class="java.lang.String" itemvalue="yarl" />
|
| 76 |
+
<item index="68" class="java.lang.String" itemvalue="webencodings" />
|
| 77 |
+
<item index="69" class="java.lang.String" itemvalue="ujson" />
|
| 78 |
+
<item index="70" class="java.lang.String" itemvalue="absl-py" />
|
| 79 |
+
<item index="71" class="java.lang.String" itemvalue="protobuf" />
|
| 80 |
+
<item index="72" class="java.lang.String" itemvalue="joblib" />
|
| 81 |
+
<item index="73" class="java.lang.String" itemvalue="threadpoolctl" />
|
| 82 |
+
<item index="74" class="java.lang.String" itemvalue="huggingface-hub" />
|
| 83 |
+
<item index="75" class="java.lang.String" itemvalue="opt-einsum" />
|
| 84 |
+
<item index="76" class="java.lang.String" itemvalue="python-dotenv" />
|
| 85 |
+
<item index="77" class="java.lang.String" itemvalue="h11" />
|
| 86 |
+
<item index="78" class="java.lang.String" itemvalue="cycler" />
|
| 87 |
+
<item index="79" class="java.lang.String" itemvalue="gast" />
|
| 88 |
+
<item index="80" class="java.lang.String" itemvalue="httptools" />
|
| 89 |
+
<item index="81" class="java.lang.String" itemvalue="frozenlist" />
|
| 90 |
+
<item index="82" class="java.lang.String" itemvalue="mkl" />
|
| 91 |
+
<item index="83" class="java.lang.String" itemvalue="semantic-version" />
|
| 92 |
+
<item index="84" class="java.lang.String" itemvalue="safetensors" />
|
| 93 |
+
<item index="85" class="java.lang.String" itemvalue="oauthlib" />
|
| 94 |
+
<item index="86" class="java.lang.String" itemvalue="keras" />
|
| 95 |
+
<item index="87" class="java.lang.String" itemvalue="pyparsing" />
|
| 96 |
+
<item index="88" class="java.lang.String" itemvalue="Markdown" />
|
| 97 |
+
<item index="89" class="java.lang.String" itemvalue="sympy" />
|
| 98 |
+
<item index="90" class="java.lang.String" itemvalue="xxhash" />
|
| 99 |
+
<item index="91" class="java.lang.String" itemvalue="tokenizers" />
|
| 100 |
+
<item index="92" class="java.lang.String" itemvalue="dnspython" />
|
| 101 |
+
<item index="93" class="java.lang.String" itemvalue="h5py" />
|
| 102 |
+
<item index="94" class="java.lang.String" itemvalue="wrapt" />
|
| 103 |
+
<item index="95" class="java.lang.String" itemvalue="kiwisolver" />
|
| 104 |
+
<item index="96" class="java.lang.String" itemvalue="orjson" />
|
| 105 |
+
<item index="97" class="java.lang.String" itemvalue="tensorflow-intel" />
|
| 106 |
+
<item index="98" class="java.lang.String" itemvalue="fonttools" />
|
| 107 |
+
<item index="99" class="java.lang.String" itemvalue="tbb" />
|
| 108 |
+
<item index="100" class="java.lang.String" itemvalue="peft" />
|
| 109 |
+
<item index="101" class="java.lang.String" itemvalue="charset-normalizer" />
|
| 110 |
+
<item index="102" class="java.lang.String" itemvalue="ffmpy" />
|
| 111 |
+
<item index="103" class="java.lang.String" itemvalue="mkl-service" />
|
| 112 |
+
<item index="104" class="java.lang.String" itemvalue="shellingham" />
|
| 113 |
+
<item index="105" class="java.lang.String" itemvalue="tensorflow-addons" />
|
| 114 |
+
<item index="106" class="java.lang.String" itemvalue="wcwidth" />
|
| 115 |
+
<item index="107" class="java.lang.String" itemvalue="llvmlite" />
|
| 116 |
+
<item index="108" class="java.lang.String" itemvalue="typeguard" />
|
| 117 |
+
<item index="109" class="java.lang.String" itemvalue="requests-oauthlib" />
|
| 118 |
+
<item index="110" class="java.lang.String" itemvalue="tomlkit" />
|
| 119 |
+
<item index="111" class="java.lang.String" itemvalue="fastapi-cli" />
|
| 120 |
+
<item index="112" class="java.lang.String" itemvalue="typer" />
|
| 121 |
+
<item index="113" class="java.lang.String" itemvalue="importlib_resources" />
|
| 122 |
+
<item index="114" class="java.lang.String" itemvalue="intel-openmp" />
|
| 123 |
+
<item index="115" class="java.lang.String" itemvalue="dill" />
|
| 124 |
+
<item index="116" class="java.lang.String" itemvalue="pooch" />
|
| 125 |
+
<item index="117" class="java.lang.String" itemvalue="fastapi" />
|
| 126 |
+
<item index="118" class="java.lang.String" itemvalue="aiohttp" />
|
| 127 |
+
<item index="119" class="java.lang.String" itemvalue="grpcio" />
|
| 128 |
+
<item index="120" class="java.lang.String" itemvalue="aiosignal" />
|
| 129 |
+
<item index="121" class="java.lang.String" itemvalue="google-auth" />
|
| 130 |
+
<item index="122" class="java.lang.String" itemvalue="tensorflow-cpu" />
|
| 131 |
+
<item index="123" class="java.lang.String" itemvalue="jupyter" />
|
| 132 |
+
<item index="124" class="java.lang.String" itemvalue="torchaudio_augmentations" />
|
| 133 |
+
<item index="125" class="java.lang.String" itemvalue="bert_score" />
|
| 134 |
+
<item index="126" class="java.lang.String" itemvalue="rouge_score" />
|
| 135 |
+
</list>
|
| 136 |
+
</value>
|
| 137 |
+
</option>
|
| 138 |
+
</inspection_tool>
|
| 139 |
+
</profile>
|
| 140 |
+
</component>
|
.idea/inspectionProfiles/profiles_settings.xml
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
<component name="InspectionProjectProfileManager">
|
| 2 |
+
<settings>
|
| 3 |
+
<option name="USE_PROJECT_PROFILE" value="false" />
|
| 4 |
+
<version value="1.0" />
|
| 5 |
+
</settings>
|
| 6 |
+
</component>
|
.idea/misc.xml
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
<?xml version="1.0" encoding="UTF-8"?>
|
| 2 |
+
<project version="4">
|
| 3 |
+
<component name="Black">
|
| 4 |
+
<option name="sdkName" value="Testing" />
|
| 5 |
+
</component>
|
| 6 |
+
<component name="ProjectRootManager" version="2" project-jdk-name="Testing" project-jdk-type="Python SDK" />
|
| 7 |
+
</project>
|
.idea/modules.xml
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
<?xml version="1.0" encoding="UTF-8"?>
|
| 2 |
+
<project version="4">
|
| 3 |
+
<component name="ProjectModuleManager">
|
| 4 |
+
<modules>
|
| 5 |
+
<module fileurl="file://$PROJECT_DIR$/.idea/ThetaM2V.iml" filepath="$PROJECT_DIR$/.idea/ThetaM2V.iml" />
|
| 6 |
+
</modules>
|
| 7 |
+
</component>
|
| 8 |
+
</project>
|
.idea/vcs.xml
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
<?xml version="1.0" encoding="UTF-8"?>
|
| 2 |
+
<project version="4">
|
| 3 |
+
<component name="VcsDirectoryMappings">
|
| 4 |
+
<mapping directory="" vcs="Git" />
|
| 5 |
+
</component>
|
| 6 |
+
</project>
|
app.py
CHANGED
|
@@ -14,7 +14,7 @@ import torch
|
|
| 14 |
physical_devices = tf.config.experimental.list_physical_devices('GPU')
|
| 15 |
if len(physical_devices) > 0:
|
| 16 |
tf.config.experimental.set_memory_growth(physical_devices[0], True)
|
| 17 |
-
device = "
|
| 18 |
|
| 19 |
@st.cache_resource
|
| 20 |
def load_text_model():
|
|
|
|
| 14 |
physical_devices = tf.config.experimental.list_physical_devices('GPU')
|
| 15 |
if len(physical_devices) > 0:
|
| 16 |
tf.config.experimental.set_memory_growth(physical_devices[0], True)
|
| 17 |
+
device = "cpu"
|
| 18 |
|
| 19 |
@st.cache_resource
|
| 20 |
def load_text_model():
|