h-siyuan commited on
Commit
b226556
·
verified ·
1 Parent(s): 2c8e1ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -85,10 +85,9 @@ def run_showui(image, query):
85
 
86
  # Prepare inputs for the model
87
 
88
- global model, processor
89
 
90
  model = model.to("cuda")
91
- processor = processor.to("cuda")
92
 
93
  text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
94
  image_inputs, video_inputs = process_vision_info(messages)
 
85
 
86
  # Prepare inputs for the model
87
 
88
+ global model
89
 
90
  model = model.to("cuda")
 
91
 
92
  text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
93
  image_inputs, video_inputs = process_vision_info(messages)