fantaxy commited on
Commit
5ef4699
1 Parent(s): e683bf1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ torch.cuda.empty_cache()
34
  if torch.cuda.is_available():
35
  torch.cuda.set_per_process_memory_fraction(0.7) # Use only 70% of GPU memory
36
 
37
- huggingface_token = os.getenv("HF_TOKEN")
38
 
39
  model_path = snapshot_download(
40
  repo_id="black-forest-labs/FLUX.1-dev",
 
34
  if torch.cuda.is_available():
35
  torch.cuda.set_per_process_memory_fraction(0.7) # Use only 70% of GPU memory
36
 
37
+ huggingface_token = os.getenv("HUGGINFACE_TOKEN")
38
 
39
  model_path = snapshot_download(
40
  repo_id="black-forest-labs/FLUX.1-dev",