[Admin maintenance] Support new ZeroGPU hardware

#5
by multimodalart HF Staff - opened
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +8 -7
  3. requirements.txt +2 -9
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 🌍
4
  colorFrom: pink
5
  colorTo: pink
6
  sdk: gradio
7
- sdk_version: 4.44.0
8
  app_file: app.py
9
  pinned: false
10
  short_description: Official Demo of Lotus (https://lotus3d.github.io/)
 
4
  colorFrom: pink
5
  colorTo: pink
6
  sdk: gradio
7
+ sdk_version: 5.49.1
8
  app_file: app.py
9
  pinned: false
10
  short_description: Official Demo of Lotus (https://lotus3d.github.io/)
app.py CHANGED
@@ -1,22 +1,23 @@
1
- from gradio_imageslider import ImageSlider
2
- import functools
3
  import os
 
4
  import tempfile
 
 
 
 
5
  import diffusers
6
  import gradio as gr
 
 
7
  import imageio as imageio
8
  import numpy as np
9
- import spaces
10
  import torch as torch
11
  from PIL import Image
12
  from tqdm import tqdm
13
- from pathlib import Path
14
- import gradio
15
- from gradio.utils import get_cache_folder
16
  from infer import lotus, lotus_video
17
  import transformers
18
 
19
- transformers.utils.move_cache()
20
 
21
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
22
 
 
 
 
1
  import os
2
+ import functools
3
  import tempfile
4
+ from pathlib import Path
5
+
6
+ import spaces
7
+
8
  import diffusers
9
  import gradio as gr
10
+ import gradio
11
+ from gradio_imageslider import ImageSlider
12
  import imageio as imageio
13
  import numpy as np
 
14
  import torch as torch
15
  from PIL import Image
16
  from tqdm import tqdm
 
 
 
17
  from infer import lotus, lotus_video
18
  import transformers
19
 
20
+ # transformers.utils.move_cache() removed in transformers 4.58+
21
 
22
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
23
 
requirements.txt CHANGED
@@ -1,17 +1,13 @@
1
  numpy==1.26.4
2
 
3
- torch==2.5.1 --index-url https://download.pytorch.org/whl/cu121
4
- torchvision==0.20.1 --index-url https://download.pytorch.org/whl/cu121
5
- torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu121
6
-
7
- diffusers==0.28.0
8
  accelerate>=0.16.0
9
  transformers>=4.25.1
10
  datasets==2.21.0
11
  ftfy==6.2.3
12
  tensorboard==2.17.1
13
  Jinja2==3.1.3
14
- peft==0.7.0
15
  # bitsandbytes==0.44.1
16
  geffnet==1.0.2
17
  opencv-python==4.10.0.82
@@ -22,7 +18,4 @@ tabulate==0.9.0
22
  imageio==2.35.1
23
  imageio-ffmpeg==0.5.1
24
  # spaces==0.28.3 # let the system decide
25
- gradio==4.44.0
26
  gradio-imageslider==0.0.20
27
- gradio-client==1.3.0
28
- huggingface_hub==0.25.0
 
1
  numpy==1.26.4
2
 
3
+ diffusers==0.36.0
 
 
 
 
4
  accelerate>=0.16.0
5
  transformers>=4.25.1
6
  datasets==2.21.0
7
  ftfy==6.2.3
8
  tensorboard==2.17.1
9
  Jinja2==3.1.3
10
+ peft>=0.17.0
11
  # bitsandbytes==0.44.1
12
  geffnet==1.0.2
13
  opencv-python==4.10.0.82
 
18
  imageio==2.35.1
19
  imageio-ffmpeg==0.5.1
20
  # spaces==0.28.3 # let the system decide
 
21
  gradio-imageslider==0.0.20