Texttra
/

Cityscape_Studio

template:diffusion-lora

Model card Files Files and versions

Texttra commited on Apr 7, 2025

Commit

ea43fb8

·

verified ·

1 Parent(s): 66aa5e4

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -15,7 +15,7 @@ class EndpointHandler:
         )
         # Load your LoRA weights from the repo
-        self.pipe.load_lora_weights(model_dir, weight_name="c1t3_v1.safetensors")
         # Move to GPU if available
         if torch.cuda.is_available():

         )
         # Load your LoRA weights from the repo
+        self.pipe.load_lora_weights(path, weight_name="c1t3_v1.safetensors")
         # Move to GPU if available
         if torch.cuda.is_available():