tentando com variável
Browse files
app.py
CHANGED
|
@@ -29,6 +29,10 @@ from huggingface_hub import (
|
|
| 29 |
from diffusers.utils import load_image
|
| 30 |
|
| 31 |
import spaces
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
|
| 33 |
#---if workspace = local or colab---
|
| 34 |
|
|
|
|
| 29 |
from diffusers.utils import load_image
|
| 30 |
|
| 31 |
import spaces
|
| 32 |
+
from huggingface_hub import HfApi
|
| 33 |
+
|
| 34 |
+
token = os.getenv("HF_TOKEN")
|
| 35 |
+
good_vae = AutoencoderKL.from_pretrained(base_model, subfolder="vae", torch_dtype=dtype, use_auth_token=token).to(device)
|
| 36 |
|
| 37 |
#---if workspace = local or colab---
|
| 38 |
|