simdi commited on
Commit
9d33879
1 Parent(s): ec76d40

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -1
handler.py CHANGED
@@ -30,7 +30,7 @@ WEIGHT_PAIRS = [
30
  def float_to_pair_index(f: float):
31
  length = len(WEIGHT_PAIRS)
32
  # If f is less than length, convert to integer and use directly
33
- if f < length:
34
  return int(f)
35
  # If f is greater or equal to length, assume it's a proportion of the length
36
  else:
@@ -137,6 +137,8 @@ def generate_image(pipe, inputs):
137
  styles = inputs["styles"]
138
  if isinstance(styles, str):
139
  styles = [styles]
 
 
140
  content = inputs["content"]
141
  art_scale = inputs["art_scale"]
142
 
 
30
  def float_to_pair_index(f: float):
31
  length = len(WEIGHT_PAIRS)
32
  # If f is less than length, convert to integer and use directly
33
+ if False:
34
  return int(f)
35
  # If f is greater or equal to length, assume it's a proportion of the length
36
  else:
 
137
  styles = inputs["styles"]
138
  if isinstance(styles, str):
139
  styles = [styles]
140
+ if len(styles) == 1:
141
+ styles = styles * 5
142
  content = inputs["content"]
143
  art_scale = inputs["art_scale"]
144