Spaces:
Running
on
Zero
Running
on
Zero
Fix final model compatibility
Browse files
app.py
CHANGED
|
@@ -41,7 +41,7 @@ def run_xlc(prompt, negative_prompt=None, guidance_scale=7.0, pag_scale=3.0, pag
|
|
| 41 |
seed = random.randint(0, 9007199254740991)
|
| 42 |
|
| 43 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|
| 44 |
-
image = pipe_xlc(prompt, negative_prompt=negative_prompt, guidance_scale=guidance_scale, pag_scale=pag_scale, pag_applied_layers=pag_layers, generator=generator, num_inference_steps=25).images
|
| 45 |
|
| 46 |
return image, seed
|
| 47 |
|
|
|
|
| 41 |
seed = random.randint(0, 9007199254740991)
|
| 42 |
|
| 43 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|
| 44 |
+
image = pipe_xlc(prompt, negative_prompt=negative_prompt, guidance_scale=guidance_scale, pag_scale=pag_scale, pag_applied_layers=pag_layers, generator=generator, num_inference_steps=25, num_images_per_prompt=4).images
|
| 45 |
|
| 46 |
return image, seed
|
| 47 |
|