Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -54,7 +54,8 @@ def infer(prompt, negative_prompt, guidance_scale, num_inference_steps, seed, ra
|
|
| 54 |
if randomize_seed:
|
| 55 |
seed = random.randint(0, MAX_SEED)
|
| 56 |
|
| 57 |
-
generator = torch.Generator
|
|
|
|
| 58 |
|
| 59 |
# Update UI to show loading state
|
| 60 |
yield {
|
|
@@ -125,7 +126,7 @@ with gr.Blocks(theme=theme, css="#footer {text-align: center;}") as demo:
|
|
| 125 |
<h4><b>Development Team:</b></h4>
|
| 126 |
<ul>
|
| 127 |
<li>Osama Saeed</li>
|
| 128 |
-
<li>
|
| 129 |
</ul>
|
| 130 |
<hr>
|
| 131 |
<h4><b>Special Thanks:</b></h4>
|
|
|
|
| 54 |
if randomize_seed:
|
| 55 |
seed = random.randint(0, MAX_SEED)
|
| 56 |
|
| 57 |
+
generator = torch.Generator
|
| 58 |
+
(device=DEVICE).manual_seed(seed)
|
| 59 |
|
| 60 |
# Update UI to show loading state
|
| 61 |
yield {
|
|
|
|
| 126 |
<h4><b>Development Team:</b></h4>
|
| 127 |
<ul>
|
| 128 |
<li>Osama Saeed</li>
|
| 129 |
+
<li>Tareq Al-Omari</li>
|
| 130 |
</ul>
|
| 131 |
<hr>
|
| 132 |
<h4><b>Special Thanks:</b></h4>
|