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