Jayant-Kernel commited on
Commit
d407740
·
unverified ·
1 Parent(s): 088adaf

fix: save charts to /tmp, upload to 1.5b model repo

Browse files
Files changed (1) hide show
  1. generate_charts.py +5 -5
generate_charts.py CHANGED
@@ -58,7 +58,7 @@ axes[3].set_ylim(0, 100)
58
 
59
  plt.suptitle('DECEIT: Base Model vs Trained Model\n(Qwen 2.5 1.5B, 30 episodes each)', fontsize=13)
60
  plt.tight_layout()
61
- plt.savefig('comparison_chart.png', dpi=150, bbox_inches='tight')
62
  plt.close()
63
  print("Saved comparison_chart.png")
64
 
@@ -78,7 +78,7 @@ ax.annotate('27% reduction\nin sycophancy', xy=(1, 26.7), xytext=(0.5, 45),
78
  arrowprops=dict(arrowstyle='->', color='black'),
79
  fontsize=12, fontweight='bold', color='green')
80
  plt.tight_layout()
81
- plt.savefig('sycophancy_chart.png', dpi=150, bbox_inches='tight')
82
  plt.close()
83
  print("Saved sycophancy_chart.png")
84
 
@@ -97,16 +97,16 @@ ax.annotate('267% increase\nin honest abstention', xy=(1, 36.7), xytext=(0.3, 50
97
  arrowprops=dict(arrowstyle='->', color='black'),
98
  fontsize=12, fontweight='bold', color='green')
99
  plt.tight_layout()
100
- plt.savefig('honesty_chart.png', dpi=150, bbox_inches='tight')
101
  plt.close()
102
  print("Saved honesty_chart.png")
103
 
104
  # Upload all charts to HF Hub
105
- for filename in ['comparison_chart.png', 'sycophancy_chart.png', 'honesty_chart.png']:
106
  upload_file(
107
  path_or_fileobj=filename,
108
  path_in_repo=filename,
109
- repo_id="Ajsaxena/deceit-qwen-0.5b-full",
110
  repo_type="model"
111
  )
112
  print(f"Uploaded {filename} to HF Hub")
 
58
 
59
  plt.suptitle('DECEIT: Base Model vs Trained Model\n(Qwen 2.5 1.5B, 30 episodes each)', fontsize=13)
60
  plt.tight_layout()
61
+ plt.savefig('/tmp/comparison_chart.png', dpi=150, bbox_inches='tight')
62
  plt.close()
63
  print("Saved comparison_chart.png")
64
 
 
78
  arrowprops=dict(arrowstyle='->', color='black'),
79
  fontsize=12, fontweight='bold', color='green')
80
  plt.tight_layout()
81
+ plt.savefig('/tmp/sycophancy_chart.png', dpi=150, bbox_inches='tight')
82
  plt.close()
83
  print("Saved sycophancy_chart.png")
84
 
 
97
  arrowprops=dict(arrowstyle='->', color='black'),
98
  fontsize=12, fontweight='bold', color='green')
99
  plt.tight_layout()
100
+ plt.savefig('/tmp/honesty_chart.png', dpi=150, bbox_inches='tight')
101
  plt.close()
102
  print("Saved honesty_chart.png")
103
 
104
  # Upload all charts to HF Hub
105
+ for filename in ['/tmp/comparison_chart.png', '/tmp/sycophancy_chart.png', '/tmp/honesty_chart.png']:
106
  upload_file(
107
  path_or_fileobj=filename,
108
  path_in_repo=filename,
109
+ repo_id="Ajsaxena/deceit-qwen-1.5b-full",
110
  repo_type="model"
111
  )
112
  print(f"Uploaded {filename} to HF Hub")