haodongli commited on
Commit
b149a30
·
1 Parent(s): be3970e

zerogpu update

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +8 -1
  3. requirements.txt +2 -9
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 🚀
4
  colorFrom: blue
5
  colorTo: indigo
6
  sdk: gradio
7
- sdk_version: 4.44.0
8
  app_file: app.py
9
  pinned: false
10
  license: mit
 
4
  colorFrom: blue
5
  colorTo: indigo
6
  sdk: gradio
7
+ sdk_version: 5.49.1
8
  app_file: app.py
9
  pinned: false
10
  license: mit
app.py CHANGED
@@ -1,9 +1,16 @@
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
@@ -16,7 +23,7 @@ from gradio.utils import get_cache_folder
16
  from infer import lotus, lotus_video
17
  import transformers
18
 
19
- transformers.utils.move_cache()
20
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
21
 
22
  def infer(path_input, seed):
 
1
  from gradio_imageslider import ImageSlider
2
  import functools
3
  import os
4
+ import functools
5
  import tempfile
6
+ from pathlib import Path
7
+
8
+ import spaces
9
+
10
  import diffusers
11
  import gradio as gr
12
+ import gradio
13
+ from gradio_imageslider import ImageSlider
14
  import imageio as imageio
15
  import numpy as np
16
  import spaces
 
23
  from infer import lotus, lotus_video
24
  import transformers
25
 
26
+ # transformers.utils.move_cache() removed in transformers 4.58+
27
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
28
 
29
  def infer(path_input, seed):
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