Fucius commited on
Commit
b953383
1 Parent(s): b2f4caf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -26,13 +26,12 @@ from transformers import DPTFeatureExtractor, DPTForDepthEstimation
26
  from controlnet_aux import OpenposeDetector
27
  from controlnet_aux.open_pose.body import Body
28
 
29
- try:
30
- from inference.models import YOLOWorld
31
- from src.efficientvit.models.efficientvit.sam import EfficientViTSamPredictor
32
- from src.efficientvit.sam_model_zoo import create_sam_model
33
- import supervision as sv
34
- except:
35
- print("YoloWorld can not be load")
36
 
37
  try:
38
  from groundingdino.models import build_model
@@ -133,7 +132,7 @@ def sample_image(pipe,
133
  generator=None,
134
  concept_models=None,
135
  num_inference_steps=50,
136
- guidance_scale=7.5,
137
  controller=None,
138
  face_app=None,
139
  image=None,
@@ -660,7 +659,7 @@ def main(device, segment_type):
660
 
661
  def parse_args():
662
  parser = argparse.ArgumentParser('', add_help=False)
663
- parser.add_argument('--pretrained_model', default='wangqixun/YamerMIX_v8', type=str)
664
  parser.add_argument('--controlnet_path', default='./checkpoint/InstantID/ControlNetModel', type=str)
665
  parser.add_argument('--face_adapter_path', default='./checkpoint/InstantID/ip-adapter.bin', type=str)
666
  parser.add_argument('--openpose_checkpoint', default='thibaud/controlnet-openpose-sdxl-1.0', type=str)
 
26
  from controlnet_aux import OpenposeDetector
27
  from controlnet_aux.open_pose.body import Body
28
 
29
+
30
+ from inference.models import YOLOWorld
31
+ from src.efficientvit.models.efficientvit.sam import EfficientViTSamPredictor
32
+ from src.efficientvit.sam_model_zoo import create_sam_model
33
+ import supervision as sv
34
+
 
35
 
36
  try:
37
  from groundingdino.models import build_model
 
132
  generator=None,
133
  concept_models=None,
134
  num_inference_steps=50,
135
+ guidance_scale=3.0,
136
  controller=None,
137
  face_app=None,
138
  image=None,
 
659
 
660
  def parse_args():
661
  parser = argparse.ArgumentParser('', add_help=False)
662
+ parser.add_argument('--pretrained_model', default='stablediffusionapi/realism-engine-sdxl-v30', type=str)
663
  parser.add_argument('--controlnet_path', default='./checkpoint/InstantID/ControlNetModel', type=str)
664
  parser.add_argument('--face_adapter_path', default='./checkpoint/InstantID/ip-adapter.bin', type=str)
665
  parser.add_argument('--openpose_checkpoint', default='thibaud/controlnet-openpose-sdxl-1.0', type=str)