Update app_diffusers.py
Browse files- app_diffusers.py +1 -1
app_diffusers.py
CHANGED
|
@@ -27,7 +27,7 @@ pipe = OvisImagePipeline.from_pretrained(
|
|
| 27 |
)
|
| 28 |
pipe.to(device)
|
| 29 |
|
| 30 |
-
@spaces.GPU(duration=75)
|
| 31 |
def generate(prompt, img_height=1024, img_width=1024, seed=42, steps=50, guidance_scale=5.0):
|
| 32 |
print(f'inference with prompt : {prompt}, size: {img_height}x{img_width}, seed : {seed}, step : {steps}, cfg : {guidance_scale}')
|
| 33 |
generator = torch.Generator().manual_seed(seed)
|
|
|
|
| 27 |
)
|
| 28 |
pipe.to(device)
|
| 29 |
|
| 30 |
+
# @spaces.GPU(duration=75)
|
| 31 |
def generate(prompt, img_height=1024, img_width=1024, seed=42, steps=50, guidance_scale=5.0):
|
| 32 |
print(f'inference with prompt : {prompt}, size: {img_height}x{img_width}, seed : {seed}, step : {steps}, cfg : {guidance_scale}')
|
| 33 |
generator = torch.Generator().manual_seed(seed)
|