Surn commited on
Commit
7418a86
·
1 Parent(s): bea8151

Update version collection

Browse files
Files changed (3) hide show
  1. app.py +3 -4
  2. modules/version_info.py +31 -37
  3. requirements.txt +2 -1
app.py CHANGED
@@ -67,10 +67,9 @@ with gr.Blocks(css_paths="style_20250314.css", title="3D viewer", theme='Surn/Be
67
 
68
  with gr.Row():
69
  upload_btn = gr.UploadButton(
70
- "Upload",
71
- file_types=[".glb", ".gltf", ".obj", ".png", ".jpg", ".ply"],
72
- file_count="multiple",
73
- elem_id="upload_btn", key="upload_btn",
74
  )
75
  with gr.Row():
76
  gr.HTML(value=getVersions(), visible=True, elem_id="versions")
 
67
 
68
  with gr.Row():
69
  upload_btn = gr.UploadButton(
70
+ "Upload", elem_id="upload_btn", key="upload_btn",
71
+ file_count="multiple",
72
+ file_types=[".glb", ".gltf", ".obj", ".png", ".jpg", ".ply"]
 
73
  )
74
  with gr.Row():
75
  gr.HTML(value=getVersions(), visible=True, elem_id="versions")
modules/version_info.py CHANGED
@@ -14,12 +14,12 @@ def commit_hash():
14
  except Exception:
15
  return "<none>"
16
 
17
- def get_xformers_version():
18
- try:
19
- import xformers
20
- return xformers.__version__
21
- except Exception:
22
- return "<none>"
23
  def get_transformers_version():
24
  try:
25
  import transformers
@@ -27,30 +27,30 @@ def get_transformers_version():
27
  except Exception:
28
  return "<none>"
29
 
30
- def get_accelerate_version():
31
- try:
32
- import accelerate
33
- return accelerate.__version__
34
- except Exception:
35
- return "<none>"
36
- def get_safetensors_version():
37
- try:
38
- import safetensors
39
- return safetensors.__version__
40
- except Exception:
41
- return "<none>"
42
- def get_diffusers_version():
43
- try:
44
- import diffusers
45
- return diffusers.__version__
46
- except Exception:
47
- return "<none>"
48
- def get_open3d_version():
49
- try:
50
- import open3d
51
- return f"{open3d.__version__} cuda:{open3d.core.cuda.is_available()}"
52
- except Exception:
53
- return "<none>"
54
 
55
  def get_torch_info():
56
  from torch import __version__ as torch_version_, version, cuda, backends
@@ -109,19 +109,13 @@ def versions_html():
109
  &#x2000;•&#x2000;
110
  torch: {torch_version_}
111
  &#x2000;•&#x2000;
112
- diffusers: {get_diffusers_version()}
113
- &#x2000;•&#x2000;
114
  transformers: {get_transformers_version()}
115
  &#x2000;•&#x2000;
116
- safetensors: {get_safetensors_version()}
117
- &#x2000;•&#x2000;
118
- open3d: {get_open3d_version()}
119
- &#x2000;•&#x2000;
120
  gradio: {gr.__version__}
121
  &#x2000;•&#x2000;
122
  {toggle_dark_link}
123
  <br>
124
- Full GPU Info:{get_torch_info()}
125
  """
126
  del torch_version_
127
  return v_html
 
14
  except Exception:
15
  return "<none>"
16
 
17
+ # def get_xformers_version():
18
+ # try:
19
+ # import xformers
20
+ # return xformers.__version__
21
+ # except Exception:
22
+ # return "<none>"
23
  def get_transformers_version():
24
  try:
25
  import transformers
 
27
  except Exception:
28
  return "<none>"
29
 
30
+ # def get_accelerate_version():
31
+ # try:
32
+ # import accelerate
33
+ # return accelerate.__version__
34
+ # except Exception:
35
+ # return "<none>"
36
+ # def get_safetensors_version():
37
+ # try:
38
+ # import safetensors
39
+ # return safetensors.__version__
40
+ # except Exception:
41
+ # return "<none>"
42
+ # def get_diffusers_version():
43
+ # try:
44
+ # import diffusers
45
+ # return diffusers.__version__
46
+ # except Exception:
47
+ # return "<none>"
48
+ # def get_open3d_version():
49
+ # try:
50
+ # import open3d
51
+ # return f"{open3d.__version__} cuda:{open3d.core.cuda.is_available()}"
52
+ # except Exception:
53
+ # return "<none>"
54
 
55
  def get_torch_info():
56
  from torch import __version__ as torch_version_, version, cuda, backends
 
109
  &#x2000;•&#x2000;
110
  torch: {torch_version_}
111
  &#x2000;•&#x2000;
 
 
112
  transformers: {get_transformers_version()}
113
  &#x2000;•&#x2000;
 
 
 
 
114
  gradio: {gr.__version__}
115
  &#x2000;•&#x2000;
116
  {toggle_dark_link}
117
  <br>
118
+ Full GPU Info:
119
  """
120
  del torch_version_
121
  return v_html
requirements.txt CHANGED
@@ -1 +1,2 @@
1
- torch --extra-index-url https://download.pytorch.org/whl/cu124
 
 
1
+ torch==2.6.0 --extra-index-url https://download.pytorch.org/whl/cu124
2
+ transformers