mygyasir commited on
Commit
caa3e25
1 Parent(s): 803e17b

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +1 -0
demo.py CHANGED
@@ -65,6 +65,7 @@ class SdmCompressionDemo:
65
  def infer_original_model(self, text, negative, guidance_scale, steps, seed):
66
  print(f"=== ORIG model --- seed {seed}")
67
  if self.check_invalid_input(text):
 
68
  return None, "Please enter the input prompt.", None
69
  output_image, nsfw_detected, test_time = self.generate_image(self.pipe_original,
70
  text, negative, guidance_scale, steps, seed)
 
65
  def infer_original_model(self, text, negative, guidance_scale, steps, seed):
66
  print(f"=== ORIG model --- seed {seed}")
67
  if self.check_invalid_input(text):
68
+ print('hello')
69
  return None, "Please enter the input prompt.", None
70
  output_image, nsfw_detected, test_time = self.generate_image(self.pipe_original,
71
  text, negative, guidance_scale, steps, seed)