Spaces:
Running
on
Zero
Running
on
Zero
Update app_with_diffusers.py
Browse files- app_with_diffusers.py +1 -1
app_with_diffusers.py
CHANGED
@@ -80,7 +80,7 @@ def infer(prompt, input_image, steps=30, cfg_scale=7.0, guidance_end=1.0,
|
|
80 |
low_quality_image = Image.open(input_image).convert("RGB")
|
81 |
|
82 |
lq = [resize_img(low_quality_image, size=(width, height))]
|
83 |
-
generator = torch.Generator(device=
|
84 |
timesteps = [
|
85 |
i * (1000//steps) + pipe.scheduler.config.steps_offset for i in range(0, steps)
|
86 |
]
|
|
|
80 |
low_quality_image = Image.open(input_image).convert("RGB")
|
81 |
|
82 |
lq = [resize_img(low_quality_image, size=(width, height))]
|
83 |
+
generator = torch.Generator(device='cuda').manual_seed(seed)
|
84 |
timesteps = [
|
85 |
i * (1000//steps) + pipe.scheduler.config.steps_offset for i in range(0, steps)
|
86 |
]
|