Badr AlKhamissi commited on
Commit
74bceb0
1 Parent(s): 22cbcea

cast tone loss params to int

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -102,9 +102,9 @@ def set_config(semantic_concept, word, prompt_suffix, font_name, num_steps, seed
102
  cfg.seed = int(seed)
103
  cfg.num_iter = num_steps
104
  cfg.batch_size = 1
105
- cfg.loss.tone.dist_loss_weight = dist_loss_weight
106
- cfg.loss.tone.pixel_dist_kernel_blur = pixel_dist_kernel_blur
107
- cfg.loss.tone.pixel_dist_sigma = pixel_dist_sigma
108
  cfg.loss.conformal.angeles_w = angeles_w
109
 
110
  # if ' ' in cfg.word:
 
102
  cfg.seed = int(seed)
103
  cfg.num_iter = num_steps
104
  cfg.batch_size = 1
105
+ cfg.loss.tone.dist_loss_weight = int(dist_loss_weight)
106
+ cfg.loss.tone.pixel_dist_kernel_blur = int(pixel_dist_kernel_blur)
107
+ cfg.loss.tone.pixel_dist_sigma = int(pixel_dist_sigma)
108
  cfg.loss.conformal.angeles_w = angeles_w
109
 
110
  # if ' ' in cfg.word: