peterpeter8585 commited on
Commit
966ab30
1 Parent(s): f2b4626

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -7,9 +7,8 @@ import torch
7
  import transformers
8
  transformers.utils.move_cache()
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
- from huggingface_hub import HfFolder
11
- password1 = HfFolder.get_secret("password")
12
-
13
  if torch.cuda.is_available():
14
  torch.cuda.max_memory_allocated(device=device)
15
  pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
 
7
  import transformers
8
  transformers.utils.move_cache()
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
+ import os
11
+ os.environ["password"]
 
12
  if torch.cuda.is_available():
13
  torch.cuda.max_memory_allocated(device=device)
14
  pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)