ybelkada commited on
Commit
3e3cc54
β€’
1 Parent(s): 5c7822d

change cuda + description

Browse files
Files changed (2) hide show
  1. models/model_settings.py +1 -1
  2. utils/constants.py +3 -6
models/model_settings.py CHANGED
@@ -97,6 +97,6 @@ STYLEGAN_TRUNCATION_LAYERS = 8 # 0 means no truncation
97
  STYLEGAN_RANDOMIZE_NOISE = False
98
 
99
  # Settings for model running.
100
- USE_CUDA = True
101
 
102
  MAX_IMAGES_ON_DEVICE = 8
 
97
  STYLEGAN_RANDOMIZE_NOISE = False
98
 
99
  # Settings for model running.
100
+ USE_CUDA = False
101
 
102
  MAX_IMAGES_ON_DEVICE = 8
utils/constants.py CHANGED
@@ -9,7 +9,7 @@ VALID_CHOICES = [
9
  "Smiling",
10
  "Gray_Hair",
11
  ]
12
- ENABLE_GPU = True
13
  MODEL_NAME = "stylegan_ffhq"
14
  OUTPUT_LIST = [
15
  gr.outputs.Image(type="pil", label="Generated Images"),
@@ -26,13 +26,13 @@ OUTPUT_LIST = [
26
 
27
 
28
  # ⭕ This method is biased on the data it has been trained for attribute recognition. E.g. if you decide to modify the "Bald" attribute on Female faces, the method will turn it into Male faces. Future work may focus more on this direction to try to have unbiased results of the modifications.
29
-
 
30
  # </center>
31
  # </p>
32
  # """
33
  description = """
34
  <p>
35
- <center>
36
  This is an interactive demo of an extension of the <a href="https://genforce.github.io/interfacegan/">CVPR2020 InterfaceGAN paper</a>, by adding other attributes such as Hat, Bald, etc. to the generated images. Here is a step-by-step guide to use this interface:
37
  <ul>
38
  <li>🌾 Select the Random seed you want to use to generate the images</li>
@@ -40,10 +40,7 @@ This is an interactive demo of an extension of the <a href="https://genforce.git
40
  <li>πŸ›  Select the coefficient of modification (higher value means stronger modification)</li>
41
  <li>πŸ”₯ Submit!</li>
42
  </ul>
43
- Check the <a href="https://github.com/genforce/interfacegan">original repo</a> as well as the <a href="https://github.com/younesbelkada/interfacegan">extended version of the work</a>.
44
  &#x2B55; This method is biased on the data it has been trained for attribute recognition. E.g. if you decide to modify the "Bald" attribute on Female faces, the method will turn it into Male faces. Future work may focus more on this direction to try to have unbiased results of the modifications.
45
-
46
- </center>
47
  </p>
48
  """
49
  css = """
 
9
  "Smiling",
10
  "Gray_Hair",
11
  ]
12
+ ENABLE_GPU = False
13
  MODEL_NAME = "stylegan_ffhq"
14
  OUTPUT_LIST = [
15
  gr.outputs.Image(type="pil", label="Generated Images"),
 
26
 
27
 
28
  # &#x2B55; This method is biased on the data it has been trained for attribute recognition. E.g. if you decide to modify the "Bald" attribute on Female faces, the method will turn it into Male faces. Future work may focus more on this direction to try to have unbiased results of the modifications.
29
+ #Check the <a href="https://github.com/genforce/interfacegan">original repo</a> as well as the <a href="https://github.com/younesbelkada/interfacegan">extended version of the work</a>.
30
+ #
31
  # </center>
32
  # </p>
33
  # """
34
  description = """
35
  <p>
 
36
  This is an interactive demo of an extension of the <a href="https://genforce.github.io/interfacegan/">CVPR2020 InterfaceGAN paper</a>, by adding other attributes such as Hat, Bald, etc. to the generated images. Here is a step-by-step guide to use this interface:
37
  <ul>
38
  <li>🌾 Select the Random seed you want to use to generate the images</li>
 
40
  <li>πŸ›  Select the coefficient of modification (higher value means stronger modification)</li>
41
  <li>πŸ”₯ Submit!</li>
42
  </ul>
 
43
  &#x2B55; This method is biased on the data it has been trained for attribute recognition. E.g. if you decide to modify the "Bald" attribute on Female faces, the method will turn it into Male faces. Future work may focus more on this direction to try to have unbiased results of the modifications.
 
 
44
  </p>
45
  """
46
  css = """