Update app_demo.py
Browse files- app_demo.py +10 -1
app_demo.py
CHANGED
|
@@ -50,7 +50,16 @@ pipe.to("cpu")
|
|
| 50 |
|
| 51 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, safety_checker= None)
|
| 52 |
prompt = "A futuristic cityscape at sunset"
|
| 53 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 54 |
image.show()
|
| 55 |
|
| 56 |
|
|
|
|
| 50 |
|
| 51 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, safety_checker= None)
|
| 52 |
prompt = "A futuristic cityscape at sunset"
|
| 53 |
+
output = pipe(
|
| 54 |
+
prompt=prompt,
|
| 55 |
+
negative_prompt="", # ← prevents added_cond_kwargs from being None
|
| 56 |
+
#num_inference_steps=50, # adjust as you like
|
| 57 |
+
#guidance_scale=7.5 # ditto
|
| 58 |
+
requires_safety_checker=False
|
| 59 |
+
|
| 60 |
+
)
|
| 61 |
+
image = output.images[0]
|
| 62 |
+
#image = pipe(prompt).images[0]
|
| 63 |
image.show()
|
| 64 |
|
| 65 |
|