Spaces:
Runtime error
Runtime error
HardWorkingStation
commited on
Commit
•
1336e83
1
Parent(s):
72a3fae
Initial commit
Browse files
app.py
CHANGED
@@ -28,7 +28,6 @@ with st.spinner(
|
|
28 |
# )
|
29 |
with open('model/stable-diffusion.bin', 'rb') as model_file:
|
30 |
pipe = pickle.load(model_file)
|
31 |
-
st.success('Model loaded. Start caching to GPU...')
|
32 |
pipe = pipe.to(device)
|
33 |
|
34 |
|
@@ -41,7 +40,7 @@ def infer(prompt, samples=2, steps=30, scale=7.5, seed=25):
|
|
41 |
[prompt] * samples,
|
42 |
num_inference_steps=steps,
|
43 |
guidance_scale=scale,
|
44 |
-
generator=generator
|
45 |
)
|
46 |
|
47 |
images = []
|
28 |
# )
|
29 |
with open('model/stable-diffusion.bin', 'rb') as model_file:
|
30 |
pipe = pickle.load(model_file)
|
|
|
31 |
pipe = pipe.to(device)
|
32 |
|
33 |
|
40 |
[prompt] * samples,
|
41 |
num_inference_steps=steps,
|
42 |
guidance_scale=scale,
|
43 |
+
generator=generator
|
44 |
)
|
45 |
|
46 |
images = []
|