witcherderivia commited on
Commit
83a5963
·
verified ·
1 Parent(s): d26022a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ from huggingface_hub import hf_hub_download
16
 
17
  pipe = QwenImagePipeline.from_pretrained(
18
  torch_dtype=torch.bfloat16,
19
- device="cuda",
20
  model_configs=[
21
  ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509",
22
  download_source='huggingface',
@@ -43,7 +43,7 @@ telestyle= hf_hub_download(repo_id="Tele-AI/TeleStyle", filename="weights/diffsy
43
  pipe.load_lora(pipe.dit, telestyle)
44
  pipe.load_lora(pipe.dit,speedup)
45
 
46
-
47
 
48
  dtype = torch.bfloat16
49
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
16
 
17
  pipe = QwenImagePipeline.from_pretrained(
18
  torch_dtype=torch.bfloat16,
19
+ device="cpu",
20
  model_configs=[
21
  ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509",
22
  download_source='huggingface',
 
43
  pipe.load_lora(pipe.dit, telestyle)
44
  pipe.load_lora(pipe.dit,speedup)
45
 
46
+ pipe.to("cuda")
47
 
48
  dtype = torch.bfloat16
49
  device = "cuda" if torch.cuda.is_available() else "cpu"