Factor Studios commited on
Commit
79e38e3
·
verified ·
1 Parent(s): 67bfb4a

Update websocket_storage.py

Browse files
Files changed (1) hide show
  1. 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': 'vram/state',
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': 'model',
322
- 'type': 'read',
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