Spaces:
Runtime error
Runtime error
Update app.py
#67
by
Venevari
- opened
app.py
CHANGED
|
@@ -55,8 +55,7 @@ if is_colab:
|
|
| 55 |
current_model.path,
|
| 56 |
torch_dtype=torch.float16,
|
| 57 |
scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler"),
|
| 58 |
-
|
| 59 |
-
)
|
| 60 |
|
| 61 |
else:
|
| 62 |
eulera = EulerAncestralDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000)
|
|
@@ -175,8 +174,7 @@ def img_to_img(model_path, prompt, neg_prompt, img, strength, guidance, steps, w
|
|
| 175 |
current_model_path,
|
| 176 |
torch_dtype=torch.float16,
|
| 177 |
scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler"),
|
| 178 |
-
|
| 179 |
-
)
|
| 180 |
else:
|
| 181 |
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
|
| 182 |
current_model_path,
|
|
|
|
| 55 |
current_model.path,
|
| 56 |
torch_dtype=torch.float16,
|
| 57 |
scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler"),
|
| 58 |
+
|
|
|
|
| 59 |
|
| 60 |
else:
|
| 61 |
eulera = EulerAncestralDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000)
|
|
|
|
| 174 |
current_model_path,
|
| 175 |
torch_dtype=torch.float16,
|
| 176 |
scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler"),
|
| 177 |
+
|
|
|
|
| 178 |
else:
|
| 179 |
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
|
| 180 |
current_model_path,
|