Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +5,6 @@ import spaces
|
|
| 5 |
import torch
|
| 6 |
from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler
|
| 7 |
from compel import Compel, ReturnedEmbeddingsType
|
| 8 |
-
from huggingface_hub import hf_hub_download
|
| 9 |
|
| 10 |
import re
|
| 11 |
|
|
@@ -225,15 +224,9 @@ if not torch.cuda.is_available():
|
|
| 225 |
MAX_SEED = np.iinfo(np.int32).max
|
| 226 |
MAX_IMAGE_SIZE = 4096
|
| 227 |
|
| 228 |
-
local_ckpt_path = hf_hub_download(
|
| 229 |
-
repo_id="Menyu/ModelFile",
|
| 230 |
-
filename="noobaiXLNAIXL_vPred10Version.safetensors",
|
| 231 |
-
revision="main"
|
| 232 |
-
)
|
| 233 |
-
|
| 234 |
if torch.cuda.is_available():
|
| 235 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
| 236 |
-
|
| 237 |
torch_dtype=torch.float16
|
| 238 |
)
|
| 239 |
scheduler_args = {"prediction_type": "v_prediction", "rescale_betas_zero_snr": True}
|
|
|
|
| 5 |
import torch
|
| 6 |
from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler
|
| 7 |
from compel import Compel, ReturnedEmbeddingsType
|
|
|
|
| 8 |
|
| 9 |
import re
|
| 10 |
|
|
|
|
| 224 |
MAX_SEED = np.iinfo(np.int32).max
|
| 225 |
MAX_IMAGE_SIZE = 4096
|
| 226 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 227 |
if torch.cuda.is_available():
|
| 228 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
| 229 |
+
"Menyu/noobaiXLNAIXL_vPred10Version",
|
| 230 |
torch_dtype=torch.float16
|
| 231 |
)
|
| 232 |
scheduler_args = {"prediction_type": "v_prediction", "rescale_betas_zero_snr": True}
|