kwilliamson commited on
Commit
8599051
·
1 Parent(s): 163cc53

changing prompting strategy

Browse files
Files changed (1) hide show
  1. models.py +3 -3
models.py CHANGED
@@ -63,7 +63,6 @@ class LineartGenerator:
63
  return combined
64
 
65
  def generate_image(self, annotated_lineart: Image.Image,
66
- prompt: str = "",
67
  num_inference_steps: int = 30) -> Image.Image:
68
  if annotated_lineart is None:
69
  raise ValueError("No annotated lineart provided!")
@@ -73,8 +72,9 @@ class LineartGenerator:
73
  generator = torch.manual_seed(self.seed)
74
 
75
  return self.pipe(
76
- prompt=prompt,
77
  image=refined_lineart,
78
  num_inference_steps=num_inference_steps,
79
- generator=generator
 
80
  ).images[0]
 
63
  return combined
64
 
65
  def generate_image(self, annotated_lineart: Image.Image,
 
66
  num_inference_steps: int = 30) -> Image.Image:
67
  if annotated_lineart is None:
68
  raise ValueError("No annotated lineart provided!")
 
72
  generator = torch.manual_seed(self.seed)
73
 
74
  return self.pipe(
75
+ prompt="A colorful image and high resolution image",
76
  image=refined_lineart,
77
  num_inference_steps=num_inference_steps,
78
+ generator=generator,
79
+ negative_prompt="monochrome, desaturated, low contrast"
80
  ).images[0]