Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,12 +4,6 @@ import numpy as np
|
|
4 |
import modin.pandas as pd
|
5 |
from PIL import Image
|
6 |
from diffusers import DiffusionPipeline
|
7 |
-
from huggingface_hub import login
|
8 |
-
import os
|
9 |
-
|
10 |
-
login(token=os.environ.get('HF_KEY'))
|
11 |
-
|
12 |
-
device = "cuda" if torch.cuda.is_available() else "cpu"
|
13 |
|
14 |
if torch.cuda.is_available():
|
15 |
PYTORCH_CUDA_ALLOC_CONF={'max_split_size_mb': 6000}
|
|
|
4 |
import modin.pandas as pd
|
5 |
from PIL import Image
|
6 |
from diffusers import DiffusionPipeline
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
|
8 |
if torch.cuda.is_available():
|
9 |
PYTORCH_CUDA_ALLOC_CONF={'max_split_size_mb': 6000}
|