Update gradio_app.py
Browse files- gradio_app.py +1 -1
gradio_app.py
CHANGED
|
@@ -109,7 +109,7 @@ def gen_mvimg(
|
|
| 109 |
):
|
| 110 |
if seed == 0:
|
| 111 |
seed = np.random.randint(1, 65535)
|
| 112 |
-
global
|
| 113 |
generator = torch.Generator(device)
|
| 114 |
generator.manual_seed(seed)
|
| 115 |
|
|
|
|
| 109 |
):
|
| 110 |
if seed == 0:
|
| 111 |
seed = np.random.randint(1, 65535)
|
| 112 |
+
global generator, device
|
| 113 |
generator = torch.Generator(device)
|
| 114 |
generator.manual_seed(seed)
|
| 115 |
|