Monke64 commited on
Commit
dc33e60
1 Parent(s): 728ab38

Changed device

Browse files
.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 = "cuda:0" if torch.cuda.is_available() else "cpu"
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():