Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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("
|
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",
|