Texttra commited on
Commit
398d54b
·
verified ·
1 Parent(s): 85d68a2

Update handler.py

Browse files
Files changed (1) hide show
  1. 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=1.0)
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=28,
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.")