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

Fix CUDA already initialized error

Browse files
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