Spaces:
Running
Running
Update fn.py
Browse files
fn.py
CHANGED
@@ -38,7 +38,9 @@ def load_model(_model = None, _vae = None, loras = []):
|
|
38 |
for lora in loras:
|
39 |
pipe.load_lora_weights(".", weight_name=lora + ".safetensors")
|
40 |
|
41 |
-
|
|
|
|
|
42 |
pipe.enable_vae_slicing()
|
43 |
|
44 |
def pil_to_webp(img):
|
|
|
38 |
for lora in loras:
|
39 |
pipe.load_lora_weights(".", weight_name=lora + ".safetensors")
|
40 |
|
41 |
+
if torch.cuda.is_available():
|
42 |
+
pipe.to("cuda")
|
43 |
+
|
44 |
pipe.enable_vae_slicing()
|
45 |
|
46 |
def pil_to_webp(img):
|