Update main.py
Browse files
main.py
CHANGED
@@ -3,6 +3,7 @@ from diffusers import StableDiffusionPipeline
|
|
3 |
import torch
|
4 |
from fastapi import FastAPI, Response
|
5 |
from fastapi.middleware.cors import CORSMiddleware
|
|
|
6 |
|
7 |
app = FastAPI()
|
8 |
|
@@ -16,7 +17,7 @@ app.add_middleware( # add the middleware
|
|
16 |
|
17 |
|
18 |
model_id = "runwayml/stable-diffusion-v1-5"
|
19 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id,
|
20 |
pipe = pipe.to("cpu")
|
21 |
pipe.enable_attention_slicing()
|
22 |
|
|
|
3 |
import torch
|
4 |
from fastapi import FastAPI, Response
|
5 |
from fastapi.middleware.cors import CORSMiddleware
|
6 |
+
from auth_token import auth_token
|
7 |
|
8 |
app = FastAPI()
|
9 |
|
|
|
17 |
|
18 |
|
19 |
model_id = "runwayml/stable-diffusion-v1-5"
|
20 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=auth_token)
|
21 |
pipe = pipe.to("cpu")
|
22 |
pipe.enable_attention_slicing()
|
23 |
|