sayakpaul HF Staff cbensimon HF Staff commited on
Commit
45c0c4e
·
verified ·
1 Parent(s): 968b96f

Fix CUDA already initialized error (#1)

Browse files

- Fix CUDA already initialized error (a693cd41308429928b5abc25ce5b1576c65b27c4)


Co-authored-by: Charles Bensimon <cbensimon@users.noreply.huggingface.co>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,11 +1,11 @@
1
  import gradio as gr
 
2
  import torch
3
  from diffusers import QwenImagePipeline
4
  from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
5
  from optimization import compile_transformer
6
  from hub_utils import _push_compiled_graph_to_hub
7
  from huggingface_hub import whoami
8
- import spaces
9
 
10
  # --- Model Loading ---
11
  dtype = torch.bfloat16
 
1
  import gradio as gr
2
+ import spaces
3
  import torch
4
  from diffusers import QwenImagePipeline
5
  from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
6
  from optimization import compile_transformer
7
  from hub_utils import _push_compiled_graph_to_hub
8
  from huggingface_hub import whoami
 
9
 
10
  # --- Model Loading ---
11
  dtype = torch.bfloat16