charliebaby2023 commited on
Commit
eaf8dd2
·
verified ·
1 Parent(s): f6e7a8e

Update app_demo.py

Browse files
Files changed (1) hide show
  1. 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
- image = pipe(prompt).images[0]
 
 
 
 
 
 
 
 
 
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