Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -49,7 +49,6 @@ birefnet = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet",
|
|
| 49 |
sd_pipe = StableDiffusionPipeline.from_pretrained(
|
| 50 |
"runwayml/stable-diffusion-v1-5",
|
| 51 |
torch_dtype=torch.float16,
|
| 52 |
-
revision="fp16",
|
| 53 |
use_auth_token=huggingface_token,
|
| 54 |
)
|
| 55 |
sd_pipe.to("cuda")
|
|
|
|
| 49 |
sd_pipe = StableDiffusionPipeline.from_pretrained(
|
| 50 |
"runwayml/stable-diffusion-v1-5",
|
| 51 |
torch_dtype=torch.float16,
|
|
|
|
| 52 |
use_auth_token=huggingface_token,
|
| 53 |
)
|
| 54 |
sd_pipe.to("cuda")
|