howard-hou commited on
Commit
544759c
·
verified ·
1 Parent(s): fef0ff8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -219,8 +219,8 @@ def pil_image_to_base64(pil_image):
219
  return base64_image
220
 
221
  image_cache = {}
222
- ln0_weight = model.w['blocks.0.ln0.weight'].to(torch.float32).to(device)
223
- ln0_bias = model.w['blocks.0.ln0.bias'].to(torch.float32).to(device)
224
  def compute_image_state(image):
225
  base64_image = pil_image_to_base64(image)
226
  if base64_image in image_cache:
 
219
  return base64_image
220
 
221
  image_cache = {}
222
+ ln0_weight = visual_rwkv.w['blocks.0.ln0.weight'].to(torch.float32).to(device)
223
+ ln0_bias = visual_rwkv.w['blocks.0.ln0.bias'].to(torch.float32).to(device)
224
  def compute_image_state(image):
225
  base64_image = pil_image_to_base64(image)
226
  if base64_image in image_cache: