Spaces:
Sleeping
Sleeping
Factor Studios
commited on
Update websocket_storage.py
Browse files- websocket_storage.py +3 -3
websocket_storage.py
CHANGED
|
@@ -235,7 +235,7 @@ class WebSocketGPUStorage:
|
|
| 235 |
|
| 236 |
# Store state in memory
|
| 237 |
operation = {
|
| 238 |
-
'operation': '
|
| 239 |
'type': 'write',
|
| 240 |
'key': state_key,
|
| 241 |
'data': state_data,
|
|
@@ -318,8 +318,8 @@ class WebSocketGPUStorage:
|
|
| 318 |
model_hash = self._calculate_model_hash(model_path)
|
| 319 |
|
| 320 |
operation = {
|
| 321 |
-
'operation': '
|
| 322 |
-
'type': '
|
| 323 |
'model_name': model_name,
|
| 324 |
'model_hash': model_hash,
|
| 325 |
'model_data': model_data
|
|
|
|
| 235 |
|
| 236 |
# Store state in memory
|
| 237 |
operation = {
|
| 238 |
+
'operation': 'store_state',
|
| 239 |
'type': 'write',
|
| 240 |
'key': state_key,
|
| 241 |
'data': state_data,
|
|
|
|
| 318 |
model_hash = self._calculate_model_hash(model_path)
|
| 319 |
|
| 320 |
operation = {
|
| 321 |
+
'operation': 'load_model',
|
| 322 |
+
'type': 'write',
|
| 323 |
'model_name': model_name,
|
| 324 |
'model_hash': model_hash,
|
| 325 |
'model_data': model_data
|