Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,15 +30,6 @@ OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./outputs")
|
|
| 30 |
|
| 31 |
from diffusers.utils import make_image_grid
|
| 32 |
|
| 33 |
-
ACCESS_TOKEN = os.environ["ACCESS_TOKEN"]
|
| 34 |
-
login(token=ACCESS_TOKEN)
|
| 35 |
-
pipeline = DiffusionPipeline.from_pretrained(
|
| 36 |
-
"Tasty-Rice/Magic_on_paper/blob/main/Magic_on_paper-SDXL-v3.safetensors",
|
| 37 |
-
trust_remote_code=True,
|
| 38 |
-
torch_dtype=torch.float16,
|
| 39 |
-
use_auth_token=ACCESS_TOKEN
|
| 40 |
-
)
|
| 41 |
-
|
| 42 |
MODEL = os.getenv(
|
| 43 |
"MODEL",
|
| 44 |
"https://huggingface.co/Tasty-Rice/Magic_on_paper/blob/main/Magic_on_paper-SDXL-v3.safetensors",
|
|
@@ -48,7 +39,6 @@ torch.backends.cudnn.deterministic = True
|
|
| 48 |
torch.backends.cudnn.benchmark = False
|
| 49 |
|
| 50 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
| 51 |
-
load_pipeline.to(device)
|
| 52 |
|
| 53 |
def load_pipeline(model_name):
|
| 54 |
vae = AutoencoderKL.from_pretrained(
|
|
|
|
| 30 |
|
| 31 |
from diffusers.utils import make_image_grid
|
| 32 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
MODEL = os.getenv(
|
| 34 |
"MODEL",
|
| 35 |
"https://huggingface.co/Tasty-Rice/Magic_on_paper/blob/main/Magic_on_paper-SDXL-v3.safetensors",
|
|
|
|
| 39 |
torch.backends.cudnn.benchmark = False
|
| 40 |
|
| 41 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
|
|
|
| 42 |
|
| 43 |
def load_pipeline(model_name):
|
| 44 |
vae = AutoencoderKL.from_pretrained(
|