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 |
|