跪请众朋友帮我找下问题出在哪里?--3

本文介绍了一个名为IEHelper的浏览器插件实现细节。该插件能够读取配置文件中的URL列表,并根据不同类型的URL进行处理。当用户导航到特定URL时,插件会发送消息给指定窗口,用于进一步的数据处理。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

// IEHlprObj.cpp : Implementation of CIEHlprObj

#include "stdafx.h"
#include "IEHelper.h"
#include "IEHlprObj.h"
#include "ExDispID.h"
#include <strstrea.h>
#include <stdio.h>
#include "ss_id.h"
#include "MyFunction.h"
#include <vector>
#include <string>
using namespace std;

#define BSC_FILE "DSC_CONFIG.INF"
const char* const pszAppName = "IEHelper";
vector <string> g_urlArr;
vector <string> g_urlSpecArr;
char syspath[MAX_PATH] = {0};
/////////////////////////////////////////////////////////////////////////////
// CIEHlprObj

//
// CIEHlprObj Methods
//

void Spilt(char* spiltString, char* strSeps, int flag)
{
 char *p = new char [strlen(spiltString)+1];
 strcpy(p, spiltString);
 char seps[32];
 strcpy(seps, strSeps);
 char *token;
 token = strtok( p, seps );
 while( token != NULL )
 {
  if(flag==0)
   g_urlArr.push_back(token);
  else
   g_urlSpecArr.push_back(token);

  token = strtok( NULL, seps );
 }
 delete p;
 p = NULL;
}

char* LoadSettingStr(char* strKey, char* defaultData, char* strFile)
{
 char buff[1024];
 ZeroMemory(buff, sizeof(buff));
 GetPrivateProfileString("HEADER", strKey, defaultData, buff, 1024 , strFile);
 return buff;
}

CIEHlprObj::CIEHlprObj()
{
 GetWindowsDirectory(syspath,MAX_PATH);
 strcat(syspath, "//");
 strcat(syspath, BSC_FILE);
 char* str = LoadSettingStr("url", "", syspath);
 Spilt(str, ",", 0);
 str = LoadSettingStr("spec_msg", "", syspath);
 Spilt(str, ",", 1);
}

CIEHlprObj::~CIEHlprObj()
{
 g_urlArr.clear();
}

BOOL CIEHlprObj::ManageConnection(enum ConnectType eConnectType)
{
 if (!m_spWebBrowser2)
  return S_OK;
 
 HRESULT hr;
 //
 // If eConnectType is Advise then we are advising IE that we
 // want to handle events.  If eConnectType is Unadvise, we are
 // telling IE that we no longer want to handle events.
 //
 CComQIPtr<IConnectionPointContainer,
  &IID_IConnectionPointContainer> spCPContainer(m_spWebBrowser2);
 
 if (spCPContainer != NULL)
 {
  CComPtr<IConnectionPoint> spConnectionPoint;
  
  hr = spCPContainer->FindConnectionPoint(DIID_DWebBrowserEvents2, &spConnectionPoint);
  if (SUCCEEDED(hr))
  {
   if (eConnectType == Advise)
   {
    //
    // Advise the client site of our desire to be handle events
    //
    hr = spConnectionPoint->Advise((IDispatch*)this, &m_dwCookie);
    if (FAILED(hr))
     ATLTRACE("/n%s: ManageConnection(): Failed to Advise/n/n", pszAppName);
   }
   else
   {
    // Remove us from the list of people interested...
    hr = spConnectionPoint->Unadvise(m_dwCookie);
    if (FAILED(hr))
     ATLTRACE("/npszAppName: ManageConnection(): Failed to Unadvise/n/n", pszAppName);
   }
  }
 }
 
 return (SUCCEEDED(hr));
}

//
// IOleObjectWithSite Methods
//
STDMETHODIMP CIEHlprObj::SetSite(IUnknown *pUnkSite)
{
 USES_CONVERSION;
 
 if (!pUnkSite)
  ATLTRACE("/nSetSite(): pUnkSite is NULL/n/n");
 else
 {
  // Query pUnkSite for the IWebBrowser2 interface.
  m_spWebBrowser2 = pUnkSite;
  
  if (m_spWebBrowser2)
  {
   // Connect to the browser in order to handle events.
   if (!ManageConnection(Advise))
    ::MessageBox(NULL, _T("Failure sinking events from IWebBrowser2"), pszAppName, MB_OK);
  }
 }
 
 return S_OK;
}


//
// IDispatch Methods
//
STDMETHODIMP CIEHlprObj::Invoke(DISPID dispidMember, REFIID riid, LCID lcid, WORD wFlags,
                                DISPPARAMS* pDispParams, VARIANT* pvarResult,
                                EXCEPINFO*  pExcepInfo,  UINT* puArgErr)
{
 USES_CONVERSION;
 char *str = NULL;
 if (!pDispParams)
  return E_INVALIDARG;
 //HGLOBAL  simpleData  =  GlobalAlloc(GPTR,512); 
 //LPBYTE  lpSmpData  =  (LPBYTE)GlobalLock(simpleData);
 HWND hwnd = FindWindow("#32770", "URLANDPROCESS");
 //if(hwnd==NULL)
 // return S_OK;
 char buff[256]={0};
 //是否为要拷贝IE中的不需要拷贝的
 bool bSpec = false;
 //
 // Get the current URL
 // 
 LPOLESTR lpURL = NULL;
 m_spWebBrowser2->get_LocationURL(&lpURL);
 int i=0;
 bool bFind = false;
 switch (dispidMember)
 {
  //
  // The parameters for this DISPID are as follows:
  // [0]: Cancel flag  - VT_BYREF|VT_BOOL
  // [1]: HTTP headers - VT_BYREF|VT_VARIANT
  // [2]: Address of HTTP POST data  - VT_BYREF|VT_VARIANT
  // [3]: Target frame name - VT_BYREF|VT_VARIANT
  // [4]: Option flags - VT_BYREF|VT_VARIANT
  // [5]: URL to navigate to - VT_BYREF|VT_VARIANT
  // [6]: An object that evaluates to the top-level or frame
  //      WebBrowser object corresponding to the event.
  //
 case DISPID_ONQUIT:
  //::MessageBox(0, "ddddd", 0, 0);
  WriteLog("C://bho.doc", "IEELPER的消息------------------");
  SendMessage(hwnd, WM_USER+116, 1, GetCurrentProcessId());
  break;

 case DISPID_BEFORENAVIGATE2:
  if (pDispParams->cArgs >= 5 && pDispParams->rgvarg[5].vt == (VT_BYREF|VT_VARIANT))
  {
            CComVariant varURL(*pDispParams->rgvarg[5].pvarVal);
            varURL.ChangeType(VT_BSTR);
   //转化要访问的网址为char *型
   str = OLE2A(varURL.bstrVal);
  }
        WriteLog("C://bho.doc", str);
 
  for(i=0; i<g_urlSpecArr.size(); i++)
  {
   if(strstr(str, g_urlSpecArr[i].c_str()))
   {
    bSpec = true;
    bFind = false;
    break;
   }
  }
  if(!bSpec)
  {
   for(i=0; i<g_urlArr.size(); i++)
   {
    if(strstr(str, g_urlArr[i].c_str()))
    {
     long id = GetCurrentProcessId();    
     sprintf(buff, "IE_ID %d ", id);
     WriteLog("C://bho.doc", "IEELPER的消息------------------");
     SendMessage(hwnd, WM_USER+104, 0, id);
     bFind = true;
    }
   }
  }
  if(!bFind)
  {
   long id = GetCurrentProcessId();
   sprintf(buff, "IE_ID %d ", id);
   WriteLog("C://bho.doc", "IEELPER的消息------------------");
   if(bSpec)
    SendMessage(hwnd, WM_USER+105, 1, id);
   else
    SendMessage(hwnd, WM_USER+105, 0, id);
  }

  break;
  
  //
  // The parameters for this DISPID:
  // [0]: URL navigated to - VT_BYREF|VT_VARIANT
  // [1]: An object that evaluates to the top-level or frame
  //      WebBrowser object corresponding to the event.
  //
 case DISPID_NAVIGATECOMPLETE2:
  
  break;

 case DISPID_DOCUMENTCOMPLETE:
  break;
  
 case DISPID_DOWNLOADBEGIN:
  break;
  
 case DISPID_DOWNLOADCOMPLETE:
  break;
  
 //case DISPID_NEWWINDOW2:
  //设置是否允许弹出窗口
  /*READYSTATE m_ReadyState;
        m_spWebBrowser2->get_ReadyState(&m_ReadyState);
        if (m_ReadyState!=READYSTATE_COMPLETE)
        {
            *pDispParams->rgvarg[0].pboolVal = TRUE;
            return S_OK;
        }
  else
  {
            *pDispParams->rgvarg[0].pboolVal = FALSE;
            return S_OK;
  }*/
     break;
  
  // The parameters for this DISPID:
  // [0]: Address of cancel flag - VT_BYREF|VT_BOOL
  //
  
 default:
  
  break;
 }
 //GlobalUnlock(simpleData); 
 return S_OK;
}

<think>嗯,用户之前问过关于AI写作工具的问题,现在他具体想了解如何部署Chinese-LLaMA-Alpaca-2来写小说,而且他自称是小白,所以需要详细且易懂的步骤。首先,我需要回顾之前给他的回答,里面提到了Chinese-LLaMA-Alpaca项目,现在他要部署的是第二版,也就是Chinese-LLaMA-Alpaca-2。 用户的需求很明确:部署这个模型来助写小说。但作为新手,可能需要从最基础的步骤开始,包括硬件要求、环境配置、模型下载、运行步骤,以及实际应用的例子。我需要确保每个步骤都清晰,避免使用过于专业的术语,或者在必要时解释术语。 首先,硬件要求方面,用户需要知道最低配置,比如显卡型号和内存大小。之前提到过RTX 3060+16G内存,这里可能需要再次确认项目的官方推荐配置。可能需要查阅GitHub仓库的README,确保信息准确。 然后是系统环境,用户可能使用的是Windows、Mac还是Linux?作为小白,可能更倾向于Windows,但项目可能更支持Linux,所以可能需要说明不同系统的注意事项,或者推荐使用Windows的WSL或者Docker。 接下来是具体的部署步骤:克隆仓库、安装依赖、下载模型、运行模型。每个步骤需要详细说明,包括可能的命令和常见错误处理。例如,安装依赖时可能需要用到pip,要确保Python版本正确,或者使用虚拟环境。 模型下载部分,用户可能需要知道从哪里下载,Hugging Face的链接是否正确,以及模型文件的大小,是否需要解压等。此外,模型微调部分可能对新手来说比较复杂,是否需要简化步骤,或者推荐使用预训练模型直接推理? 实际应用部分,需要给用户一个具体的例子,比如如何输入提示词,生成小说内容。可能需要展示一个脚本示例,并说明如何调整参数,如temperature和max_length,来影响生成结果。 此外,用户可能需要知道如何将模型集成到其他工具中,比如SillyTavern,这在之前的回答中提到过。是否需要说明如何结合使用,或者保持步骤独立? 还要考虑可能遇到的问题,比如CUDA版本不匹配,依赖安装失败,或者显存不足等。需要提前给解决方案或建议,比如降低batch size,或者使用CPU模式(虽然速度慢)。 最后,风险提示和优化建议也很重要,比如数据隐私、硬件要求高,以及如何通过微调提升效果。用户可能不知道LoRA是什么,需要简单解释。 总结下来,回答的结构应该是:硬件准备、环境配置、部署步骤、使用示例、常见问题、优化建议。每个部分要简明扼要,避免信息过载,同时提供足够的细节让小白能一步步操作。</think>明白!作为新手部署**Chinese-LLaMA-Alpaca-2**写小说,按以下步骤操作即可(含避坑指南): --- ### 🔧 **硬件准备(最低配置)** - **显卡**:NVIDIA RTX 3060(显存≥12GB) - **内存**:16GB以上 - **硬盘**:至少40GB可用空间(模型文件约30GB) --- ### 📥 **极简部署流程** #### ❶ **安装基础环境** ```bash # 适用于Windows/Mac/Linux的万能方案 conda create -n novelai python=3.10 conda activate novelai pip install torch==2.0.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 pip install -U transformers==4.33.0 accelerate sentencepiece ``` #### ❷ **下载核心文件** ```bash git clone https://github.com/ymcui/Chinese-LLaMA-Alpaca-2 cd Chinese-LLaMA-Alpaca-2 # 下载7B中文模型(新手推荐) wget https://huggingface.co/ziqingyang/chinese-alpaca-2-7b/resolve/main/chinese-alpaca-2-7b.zip unzip chinese-alpaca-2-7b.zip ``` #### ❸ **启动写作模式** 创建`novel_writer.py`文件,写入: ```python from transformers import AutoTokenizer, AutoModelForCausalLM model = AutoModelForCausalLM.from_pretrained("chinese-alpaca-2-7b") tokenizer = AutoTokenizer.from_pretrained("chinese-alpaca-2-7b") prompt = "第一章 陨落的天才\n昔日云岚宗少主萧炎,此刻正在演武场中央。三年前突然消失的斗气..." inputs = tokenizer(prompt, return_tensors="pt") outputs = model.generate(inputs.input_ids, max_length=500, temperature=0.7) print(tokenizer.decode(outputs[0], skip_special_tokens=True)) ``` #### ❹ **运行脚本** ```bash python novel_writer.py ``` --- ### 🚨 **新手必看避坑指南** 1. **显存不足报错**:在`model.generate()`中添加参数: ```python outputs = model.generate(..., pad_token_id=tokenizer.eos_token_id, max_new_tokens=300) ``` 2. **中文乱码问题**:在脚本开头添加: ```python import sys sys.stdout.reconfigure(encoding=&#39;utf-8&#39;) ``` 3. **生成内容重复**:调整`temperature=0.9`(值越大越随机) --- ### ✨ **进阶技巧:网文风格强化** 修改prompt为: ```python prompt = """[武侠修真][黄金三章节奏][主角被退婚] 用猫腻的风格描写:林家天才少年林动,因意外失去修为,在家族大比前夕被未婚妻当退婚... 要求包含:①戒指里的残魂 ②隐藏的太古血脉 ③经典打脸剧情""" ``` --- ### 📱 **懒人方案(CPU版)** 如果显卡不达标,修改加载方式: ```python model = AutoModelForCausalLM.from_pretrained("chinese-alpaca-2-7b", device_map="auto", load_in_8bit=True) ``` ⚠️ 生成速度会慢3-5倍,但能跑! --- ### 📚 **推荐训练素材** 往`./data`目录放入: - 《斗破苍穹》txt - 《诡秘之主》txt - 自己写的人物设定文档 运行微调脚本: ```bash python finetune.py --data_dir data --output_dir output ``` 需要具体调试某个生成问题,可以发你的输结果给我看~
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值