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 |
|