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 = []
|