Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ pipe.load_lora_weights("prithivMLmods/SD3.5-Large-Photorealistic-LoRA", weight_n
|
|
| 14 |
pipe.fuse_lora(lora_scale=1.0)
|
| 15 |
|
| 16 |
# 如果有GPU，转移到GPU
|
| 17 |
-
pipe.to("cuda")
|
| 18 |
|
| 19 |
# 定义图像生成函数，添加种子参数
|
| 20 |
def generate_image(prompt, seed):
|
|
|
|
| 14 |
pipe.fuse_lora(lora_scale=1.0)
|
| 15 |
|
| 16 |
# 如果有GPU，转移到GPU
|
| 17 |
+
# pipe.to("cuda")
|
| 18 |
|
| 19 |
# 定义图像生成函数，添加种子参数
|
| 20 |
def generate_image(prompt, seed):
|