Spaces:

gokaygokay
/

FLUX.1-dev-with-Captioner

Running on Zero

gokaygokay commited on Oct 27, 2024

Commit

c4c97e8

1 Parent(s): 3f0fac3

cuda_home

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ dtype = torch.bfloat16
 torch.backends.cuda.matmul.allow_tf32 = True
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
-print(subprocess.check_output(["nvcc", "--version"]).decode("utf-8"))
 login(token=huggingface_token)
 # Set up environment and cache

 torch.backends.cuda.matmul.allow_tf32 = True
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
+print(os.getenv("CUDA_HOME"))
 login(token=huggingface_token)
 # Set up environment and cache