Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 =
|
223 |
-
ln0_bias =
|
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:
|