Spaces:
Sleeping
Sleeping
Update imgGen.py
Browse files
imgGen.py
CHANGED
@@ -19,7 +19,7 @@ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, times
|
|
19 |
|
20 |
def generateTransparentImage(text):
|
21 |
# Ensure using the same inference steps as the loaded model and CFG set to 0.
|
22 |
-
image = pipe(text+', full length,
|
23 |
return image
|
24 |
|
25 |
if __name__ == "__main__":
|
|
|
19 |
|
20 |
def generateTransparentImage(text):
|
21 |
# Ensure using the same inference steps as the loaded model and CFG set to 0.
|
22 |
+
image = pipe(text+', full length, one, transparent background, vibrant', num_inference_steps=4, guidance_scale=0).images[0]
|
23 |
return image
|
24 |
|
25 |
if __name__ == "__main__":
|