Spaces:
Runtime error
Runtime error
Update image_generator.py
Browse files- image_generator.py +1 -1
image_generator.py
CHANGED
@@ -108,7 +108,7 @@ def generate_images(
|
|
108 |
print(model_path)
|
109 |
seeds = parse_range(seeds)
|
110 |
print(seeds)
|
111 |
-
seeds=[random.randint(1,
|
112 |
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
113 |
G = MyGenerator.from_pretrained(model_path)
|
114 |
os.makedirs(outdir, exist_ok=True)
|
|
|
108 |
print(model_path)
|
109 |
seeds = parse_range(seeds)
|
110 |
print(seeds)
|
111 |
+
seeds=[random.randint(1,200)]
|
112 |
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
113 |
G = MyGenerator.from_pretrained(model_path)
|
114 |
os.makedirs(outdir, exist_ok=True)
|