yangheng commited on
Commit
0da0452
·
1 Parent(s): fca804e
Files changed (3) hide show
  1. README.md +10 -0
  2. app.py +2 -1
  3. requirements.txt +1 -3
README.md CHANGED
@@ -1,3 +1,13 @@
 
 
 
 
 
 
 
 
 
 
1
  # Super Resolution Anime Diffusion
2
 
3
 
 
1
+ ---
2
+ title: Super Resolution Anime Diffusion
3
+ emoji: 🐉
4
+ colorFrom: blue
5
+ colorTo: green
6
+ sdk: gradio
7
+ app_file: app.py
8
+ pinned: false
9
+ ---
10
+
11
  # Super Resolution Anime Diffusion
12
 
13
 
app.py CHANGED
@@ -96,8 +96,9 @@ DEFAULT_NEGATIVE_PROMPT = "bad result, worst, random, invalid, inaccurate, imper
96
  magnifier = ImageMagnifier()
97
  start_time = time.time()
98
  is_colab = utils.is_google_colab()
 
99
  device = autocuda.auto_cuda()
100
- dtype = torch.float16 if device != "cpu" else torch.float32
101
 
102
  # Extract zip files if needed
103
  for z_file in findfile.find_cwd_files(and_key=['.zip'], exclude_key=['.ignore'], recursive=1):
 
96
  magnifier = ImageMagnifier()
97
  start_time = time.time()
98
  is_colab = utils.is_google_colab()
99
+ # This demo does not use GPU, if you want to use GPU, install torch with gpu bundle and uncomment this line
100
  device = autocuda.auto_cuda()
101
+ dtype = torch.float16 if device else torch.float32
102
 
103
  # Extract zip files if needed
104
  for z_file in findfile.find_cwd_files(and_key=['.zip'], exclude_key=['.ignore'], recursive=1):
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu113
2
  torch<1.13.0
3
  torchvision
4
  autocuda
@@ -22,5 +22,3 @@ realesrgan
22
  gradio
23
  numpy<2.0.0
24
  gradio-client
25
-
26
-
 
1
+ --extra-index-url https://download.pytorch.org/whl/cpu
2
  torch<1.13.0
3
  torchvision
4
  autocuda
 
22
  gradio
23
  numpy<2.0.0
24
  gradio-client