Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
|
@@ -15,7 +15,7 @@ class EndpointHandler:
|
|
| 15 |
|
| 16 |
print("Loading LoRA weights from: Texttra/Cityscape_Studio")
|
| 17 |
self.pipe.load_lora_weights("Texttra/Cityscape_Studio", weight_name="c1t3_v1.safetensors")
|
| 18 |
-
self.pipe.fuse_lora(lora_scale=
|
| 19 |
|
| 20 |
self.pipe.to("cuda" if torch.cuda.is_available() else "cpu")
|
| 21 |
print("Model initialized successfully.")
|
|
@@ -32,7 +32,7 @@ class EndpointHandler:
|
|
| 32 |
|
| 33 |
image = self.pipe(
|
| 34 |
prompt,
|
| 35 |
-
num_inference_steps=
|
| 36 |
guidance_scale=4.5
|
| 37 |
).images[0]
|
| 38 |
print("Image generated.")
|
|
|
|
| 15 |
|
| 16 |
print("Loading LoRA weights from: Texttra/Cityscape_Studio")
|
| 17 |
self.pipe.load_lora_weights("Texttra/Cityscape_Studio", weight_name="c1t3_v1.safetensors")
|
| 18 |
+
self.pipe.fuse_lora(lora_scale=0.9)
|
| 19 |
|
| 20 |
self.pipe.to("cuda" if torch.cuda.is_available() else "cpu")
|
| 21 |
print("Model initialized successfully.")
|
|
|
|
| 32 |
|
| 33 |
image = self.pipe(
|
| 34 |
prompt,
|
| 35 |
+
num_inference_steps=50,
|
| 36 |
guidance_scale=4.5
|
| 37 |
).images[0]
|
| 38 |
print("Image generated.")
|