Spaces:

fffiloni
/

sdxl-control-loras

Sleeping

fffiloni commited on Sep 21, 2023

Commit

089e6cb

1 Parent(s): 71e48de

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,8 +29,8 @@ custom_model = "fffiloni/eugene_jour_general"
 # This is where you load your trained weights
 pipe.load_lora_weights(custom_model, use_auth_token=True)
-#pipe.to("cuda")
-pipe.enable_model_cpu_offload()
 def infer(image_in, prompt):
     prompt = prompt

 # This is where you load your trained weights
 pipe.load_lora_weights(custom_model, use_auth_token=True)
+pipe.to("cuda")
+#pipe.enable_model_cpu_offload()
 def infer(image_in, prompt):
     prompt = prompt