Spaces:
Runtime error
Runtime error
Update demo.py
Browse files
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)
|