vcollos commited on
Commit
0117aaf
·
verified ·
1 Parent(s): de32120

tentando com variável

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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