Spaces:
Build error
Build error
Commit
·
ebd4549
1
Parent(s):
18f6a92
Update app.py
Browse files
app.py
CHANGED
|
@@ -58,11 +58,9 @@ current_model = models[1] if is_colab else models[0]
|
|
| 58 |
current_model_path = current_model.path
|
| 59 |
|
| 60 |
if is_colab:
|
| 61 |
-
|
| 62 |
current_model.path,
|
| 63 |
torch_dtype=torch.float16,
|
| 64 |
-
revision="fp16" if torch.cuda.is_available() else "fp32",
|
| 65 |
-
torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
|
| 66 |
scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler")
|
| 67 |
)
|
| 68 |
|
|
|
|
| 58 |
current_model_path = current_model.path
|
| 59 |
|
| 60 |
if is_colab:
|
| 61 |
+
pipe = StableDiffusionPipeline.from_pretrained(
|
| 62 |
current_model.path,
|
| 63 |
torch_dtype=torch.float16,
|
|
|
|
|
|
|
| 64 |
scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler")
|
| 65 |
)
|
| 66 |
|